Searched defs:x_dims (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | all_gather_broadcast_reorder.cc | 140 absl::Span<const int64_t> x_dims = x->shape().dimensions(); in Run() local
|
| /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/compiler/tf2tensorrt/convert/ops/ |
| D | data_format_vec_permute.cc | 68 const auto x_dims = x_input_.GetTrtDims(); in Validate() local
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| D | qconv_prepack.cpp | 521 auto x_dims = input_shape.has_value()?input_shape.value().vec():ideep::dims(); in _qconv_prepack_onednn() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1962 auto x_dims = RandomDims(2, kDefaultMaxRank); in TEST_F() local
1984 auto x_dims = RandomDims(2, kDefaultMaxRank); in TEST_F() local
2989 std::vector<int64_t> x_dims{RandomDim(v_dims[0])}; in TEST_F() local
4835 std::vector<int64_t> x_dims = RandomDims(/*min_rank=*/4, /*max_rank=*/4, in TEST_F() local
|