Searched defs:lhs_dims (Results 1 – 9 of 9) 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/core/kernels/mkl/ |
| D | mkl_batch_matmul_op.cc | 183 auto lhs_dims = TFShapeToMklDnnDims(lhs_shape); in CreateMatMulParams() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.h | 253 const int64 lhs_dims[] = {lhs.n1(), lhs.n2(), lhs.n3()}; in Concat3D() local
288 const int64 lhs_dims[] = {lhs.n1(), lhs.n2(), lhs.n3(), lhs.n4()}; in Concat4D() local
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 389 def window_padding_type_to_pad_values(padding_type, lhs_dims, rhs_dims, argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 1669 std::vector<int64> lhs_dims; in PartitionDotGroupOnBatch() local
2102 std::vector<int64> lhs_dims; in PartitionDotGroupOnContracting() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 1259 std::array<int64_t, 1> lhs_dims{static_cast<int64_t>(op.values().size())}; in GetHoistParams() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter.cc | 2183 int64 lhs_dims = hlo->operand(0)->shape().dimensions_size(); in EmitElementalDot() local
|
| D | hlo_instruction.cc | 3708 std::vector<string> lhs_dims(2 + dnums.input_spatial_dimensions().size()); in ConvolutionDimensionNumbersToString() local
|
| D | algebraic_simplifier_test.cc | 5177 std::vector<int64> lhs_dims = {2, 3, 5}; in TEST_P() local
|