Searched defs:lhs_dims (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_dense_binary_op_shared.cc | 84 const auto lhs_dims = BCast::FromShape(TensorShape(shape_vec)); in Compute() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.h | 261 const int64 lhs_dims[] = {lhs.n1(), lhs.n2(), lhs.n3()}; in Concat3D() local
296 const int64 lhs_dims[] = {lhs.n1(), lhs.n2(), lhs.n3(), lhs.n4()}; in Concat4D() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter.cc | 2124 int64 lhs_dims = hlo->operand(0)->shape().dimensions_size(); in EmitElementalDot() local
|
| D | hlo_instruction.cc | 3054 std::vector<string> lhs_dims(2 + dnums.input_spatial_dimensions().size()); in ConvolutionDimensionNumbersToString() local
|
| D | algebraic_simplifier_test.cc | 4320 std::vector<int64> lhs_dims = {1, 3, 5}; in TEST_P() local
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 250 def _convert_padding_type_to_pad_values(padding_type, lhs_dims, rhs_dims, argument
|