Searched defs:dims_b (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | batchmatmul_to_einsum.cc | 67 const int dims_b = rhs_shape.size(); in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | op_level_cost_estimator_test.cc | 123 const std::vector<int>& dims_b, in DescribeSparseTensorDenseMatMul()
139 const std::vector<int>& dims_b, in DescribeXlaEinsum()
156 const std::vector<int>& dims_b, in DescribeEinsum()
627 const std::vector<int>& dims_b) { in DescribeBatchMatMul()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | horizontal_input_fusion.cc | 68 auto dims_b = shape_b.dimensions(); in CompareShapeDimsFromLeftToRight() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| H A D | unroll_batch_matmul.cc | 172 const int dims_b = input_array_b.shape().dimensions_count(); in Run() local
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset12.py | 384 def tensordot(g: jit_utils.GraphContext, input_a, input_b, dims_a, dims_b, out=None): argument
|
| /aosp_15_r20/external/pytorch/benchmarks/static_runtime/ |
| H A D | test_static_runtime.cc | 2184 c10::List<int64_t> dims_b{0, 2, 1}; in TEST() local
|
| /aosp_15_r20/external/pytorch/test/cpp/lazy/ |
| H A D | test_lazy_ops.cpp | 3669 std::vector<int64_t> dims_b = {0, 2}; in TEST_F() local
|