Searched defs:shape_out (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | reduce_utils.h | 25 inline void RemoveSize1Dims(int* shape_out, int& out_num_dims, int* axis_out, in RemoveSize1Dims()
59 int& out_num_axis, const int* shape_in, int* shape_out, in ResolveAxis()
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | IndirectConv2dAddressPrecalculationFixture.h | 111 …TensorShape shape_out = compute_indirect_buffer_shape(shape_conv_src, DataLayout::NHWC, sh… in compute_reference() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_cross_op.cc | 711 Tensor** values_out, Tensor** shape_out, in CreateOutputTensors()
777 Tensor* shape_out; in Compute() local
848 Tensor* shape_out; in Compute() local
921 Tensor* shape_out; in Compute() local
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item_builder.cc | 95 TensorShape* shape_out) { in ReplaceUnknownShapeDim()
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/internal/ |
| D | saved_model_api_test.cc | 249 const TF_Shape* shape_out = TF_TensorSpecShape(tensor_spec_out); in TEST_P() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 891 Shape shape_out = ShapeUtil::MakeShape(C64, {16, 5}); in TEST_F() local
914 Shape shape_out = ShapeUtil::MakeShape(C64, {16, 5}); in TEST_F() local
930 Shape shape_out = ShapeUtil::MakeShape(F32, {16, 8}); in TEST_F() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties_test.cc | 788 auto shape_out = properties.GetOutputProperties("while/Exit_1").at(0).shape(); in TEST_F() local
|
| /external/pytorch/test/ |
| D | test_nn.py | 10421 def helper(shape_in, shape_out, align_corners): argument
10441 def helper(shape_in, shape_out, align_corners): argument
|