Searched defs:dims_a (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | batchmatmul_to_einsum.cc | 66 const int dims_a = lhs_shape.size(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | horizontal_input_fusion.cc | 67 auto dims_a = shape_a.dimensions(); in CompareShapeDimsFromLeftToRight() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | unroll_batch_matmul.cc | 171 const int dims_a = input_array_a.shape().dimensions_count(); in Run() local
|
| /external/pytorch/torch/onnx/ |
| D | symbolic_opset12.py | 384 def tensordot(g: jit_utils.GraphContext, input_a, input_b, dims_a, dims_b, out=None): argument
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator_test.cc | 138 OpContext DescribeXlaEinsum(const std::vector<int>& dims_a, in DescribeXlaEinsum()
155 OpContext DescribeEinsum(const std::vector<int>& dims_a, in DescribeEinsum()
626 OpContext DescribeBatchMatMul(const std::vector<int>& dims_a, in DescribeBatchMatMul()
|
| /external/pytorch/benchmarks/static_runtime/ |
| D | test_static_runtime.cc | 2180 c10::List<int64_t> dims_a{1, 0}; in TEST() local
|
| /external/pytorch/test/cpp/lazy/ |
| D | test_lazy_ops.cpp | 3668 std::vector<int64_t> dims_a = {1, 2}; in TEST_F() local
|