| /external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/ |
| D | matmul_op_benchmark.h | 69 std::array<ssize_t, 2> lhs_dims = {m, k}; in RunMatMulMlirBenchmark() local
136 std::array<ssize_t, 2> lhs_dims = {m, k}; in RunMatMulEigenBenchmark() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_batch_matmul_helper.h | 58 auto lhs_dims = TFShapeToMklDnnDims(lhs_shape); in CreateMatMulParams() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_dense_binary_op_shared.cc | 104 const auto lhs_dims = BCast::FromShape(lhs_shape); in Compute() local
|
| /external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/template_writer/cl/ |
| D | ClTemplateElementwiseBinary.cpp | 205 const auto &lhs_dims = _lhs->tensor_shape(); in get_tag_lut() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.h | 265 const int64_t lhs_dims[] = {lhs.n1(), lhs.n2(), lhs.n3()}; in Concat3D() local
300 const int64_t lhs_dims[] = {lhs.n1(), lhs.n2(), lhs.n3(), lhs.n4()}; in Concat4D() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | unroll_batch_matmul.cc | 182 const int lhs_dims = lhs_shape.size(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 357 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 | 2027 std::vector<int64_t> lhs_dims; in PartitionDotGroupOnBatch() local
2600 std::vector<int64_t> lhs_dims; in PartitionDotGroupOnContracting() local
3042 std::vector<int64_t> lhs_dims; in PrioritizeContractingDimensionsPartitioning() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 382 std::vector<int64_t> lhs_dims; member
402 std::vector<int64_t> lhs_dims; member
453 std::vector<int64_t> lhs_dims; member
1037 std::vector<int64_t> lhs_dims(batch_dims.first), rhs_dims(batch_dims.second); in ChooseBatchMatMulArguments() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 1347 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 | 2376 int64_t lhs_dims = hlo->operand(0)->shape().dimensions_size(); in EmitElementalDot() local
|
| D | hlo_instruction.cc | 4135 std::vector<std::string> lhs_dims( in ConvolutionDimensionNumbersToString() local
|
| D | algebraic_simplifier_test.cc | 5818 std::vector<int64_t> lhs_dims = {2, 3, 5}; in TEST_P() local
|