| /external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/ |
| D | matmul_op_benchmark.h | 70 std::array<ssize_t, 2> rhs_dims = {k, n}; in RunMatMulMlirBenchmark() local
137 std::array<ssize_t, 2> rhs_dims = {k, n}; in RunMatMulEigenBenchmark() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_batch_matmul_helper.h | 59 auto rhs_dims = TFShapeToMklDnnDims(rhs_shape); in CreateMatMulParams() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_dense_binary_op_shared.cc | 105 const auto rhs_dims = BCast::FromShape(dense_t->shape()); in Compute() local
|
| /external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/template_writer/cl/ |
| D | ClTemplateElementwiseBinary.cpp | 206 const auto &rhs_dims = _rhs->tensor_shape(); in get_tag_lut() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.h | 266 const int64_t rhs_dims[] = {rhs.n1(), rhs.n2(), rhs.n3()}; in Concat3D() local
301 const int64_t rhs_dims[] = {rhs.n1(), rhs.n2(), rhs.n3(), rhs.n4()}; in Concat4D() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | unroll_batch_matmul.cc | 183 const int rhs_dims = rhs_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 | 2028 std::vector<int64_t> rhs_dims; in PartitionDotGroupOnBatch() local
2601 std::vector<int64_t> rhs_dims; in PartitionDotGroupOnContracting() local
3043 std::vector<int64_t> rhs_dims; in PrioritizeContractingDimensionsPartitioning() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 383 std::vector<int64_t> rhs_dims; member
403 std::vector<int64_t> rhs_dims; member
454 std::vector<int64_t> rhs_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 | 1337 std::array<int64_t, 1> rhs_dims{static_cast<int64_t>(op.values().size())}; in GetHoistParams() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter.cc | 2377 int64_t rhs_dims = hlo->operand(1)->shape().dimensions_size(); in EmitElementalDot() local
|
| D | hlo_instruction.cc | 4146 std::vector<std::string> rhs_dims( in ConvolutionDimensionNumbersToString() local
|
| D | algebraic_simplifier_test.cc | 5819 std::vector<int64_t> rhs_dims = lhs_dims; in TEST_P() local
|