Home
last modified time | relevance | path

Searched defs:dims_b (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dbatchmatmul_to_einsum.cc66 const int dims_b = rhs_shape.size(); in matchAndRewrite() local
Dunroll_batch_matmul.cc242 const int dims_b = rhs_shape.size(); in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhorizontal_input_fusion.cc68 auto dims_b = shape_b.dimensions(); in CompareShapeDimsFromLeftToRight() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dunroll_batch_matmul.cc170 const int dims_b = input_array_b.shape().dimensions_count(); in Run() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator_test.cc122 const std::vector<int>& dims_b, in DescribeSparseTensorDenseMatMul()
138 const std::vector<int>& dims_b, in DescribeXlaEinsum()
155 const std::vector<int>& dims_b, in DescribeEinsum()
593 const std::vector<int>& dims_b) { in DescribeBatchMatMul()