Searched defs:paddings_shape (Results 1 - 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| H A D | pad_test.cc | 48 // PadOpDynamicModel m(input_shape, paddings_shape, paddings_data);
54 std::initializer_list<int> paddings_shape,
57 paddings_ = AddConstInput(TensorType_INT32, paddings, paddings_shape);
69 // PadOpDynamicModel m(input_shape, paddings_shape);
76 std::initializer_list<int> paddings_shape) {
83 BuildInterpreter({input_shape, paddings_shape});
53 PadOpConstModel(std::initializer_list<int> input_shape, std::initializer_list<int> paddings_shape, std::initializer_list<int> paddings) argument
75 PadOpDynamicModel(std::initializer_list<int> input_shape, std::initializer_list<int> paddings_shape) argument
|
| /external/tensorflow/tensorflow/core/ops/ |
| H A D | array_ops.cc | 1934 ShapeHandle paddings_shape,
1951 c->Merge(paddings_shape, c->Matrix(num_block_dims, 2), &paddings_shape));
1931 SpaceToBatchShapeHelper(InferenceContext* c, ShapeHandle input_shape, ShapeHandle block_shape_shape, const Tensor* block_shape_t, ShapeHandle paddings_shape, const Tensor* paddings_t) argument
|
Completed in 145 milliseconds