Searched defs:expected_dims (Results 1 – 9 of 9) sorted by relevance
| /external/XNNPACK/test/ |
| D | transpose-normalization-tester.h | 31 inline TransposeNormalizationTester& expected_dims(size_t expected_dims) { in expected_dims() function
36 inline size_t expected_dims() const { return this->expected_dims_; } in expected_dims() function
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | transpose_util_test.cc | 28 const gtl::ArraySlice<int64_t> expected_dims) { in TestDimensionReduction()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | callback.h | 43 absl::InlinedVector<int64_t, 4> expected_dims; member
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor_shape_test.py | 361 def testBroadcastDynamicShape(self, x_dims, y_dims, expected_dims): argument
|
| D | dynamic_ragged_shape_test.py | 1508 def testBroadcastDynamicShape(self, x_dims, y_dims, expected_dims): argument
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference_testutil.cc | 143 auto expected_dims = str_util::Split(expected, ','); in InferShapes() local
|
| /external/pytorch/torch/csrc/jit/passes/ |
| D | symbolic_shape_runtime_fusion.cpp | 539 std::vector<size_t> expected_dims; in __anon74f36cf50602() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_concat_op.cc | 497 size_t expected_dims = expected_shape.dims(); in Compute() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 148 const std::vector<int>& expected_dims, in ValidateWeights()
2407 const std::vector<OutputCType>& expected_value) { in TestConvertConst()
2445 std::vector<int> expected_dims{1}; in TestConvertConst() local
|