Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dbatchmatmul_to_einsum.cc65 const int dims_a = lhs_shape.size(); in matchAndRewrite() local
Dunroll_batch_matmul.cc241 const int dims_a = lhs_shape.size(); in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhorizontal_input_fusion.cc67 auto dims_a = shape_a.dimensions(); in CompareShapeDimsFromLeftToRight() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dunroll_batch_matmul.cc169 const int dims_a = input_array_a.shape().dimensions_count(); in Run() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator_test.cc137 OpContext DescribeXlaEinsum(const std::vector<int>& dims_a, in DescribeXlaEinsum()
154 OpContext DescribeEinsum(const std::vector<int>& dims_a, in DescribeEinsum()
592 OpContext DescribeBatchMatMul(const std::vector<int>& dims_a, in DescribeBatchMatMul()