Searched defs:subscripts (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | linalg_grad.cc | 50 absl::optional<int> EinsumGetAxisFromLabel(absl::string_view subscripts, in EinsumGetAxisFromLabel()
81 absl::string_view subscripts) { in EinsumGetBcastSubshape()
137 Output input_shape, absl::string_view subscripts) { in EinsumGetReducedSubscripts()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | linalg_grad.py | 63 def _GetAxisFromLabel(subscripts, label): argument
88 def _GetBcastSubshape(subscripts): argument
108 def _GetReducedSubscripts(reduced_label_set, input_shape, subscripts): argument
|
| /external/pytorch/test/torch_np/numpy_tests/core/ |
| D | test_einsum.py | 1041 def optimize_compare(self, subscripts, operands=None): argument
|
| /external/swiftshader/third_party/SPIRV-Tools/source/opt/ |
| D | loop_dependence_helpers.cpp | 385 std::vector<Instruction*> subscripts; in GetSubscripts() local
|
| /external/angle/third_party/spirv-tools/src/source/opt/ |
| D | loop_dependence_helpers.cpp | 385 std::vector<Instruction*> subscripts; in GetSubscripts() local
|
| /external/deqp-deps/SPIRV-Tools/source/opt/ |
| D | loop_dependence_helpers.cpp | 385 std::vector<Instruction*> subscripts; in GetSubscripts() local
|
| /external/angle/src/libANGLE/ |
| D | ProgramExecutable.cpp | 35 std::vector<unsigned int> subscripts; in IncludeSameArrayElement() local
546 std::vector<unsigned int> subscripts; in GetInterfaceBlockIndex() local
1379 std::vector<unsigned int> subscripts; in linkValidateTransformFeedback() local
1459 std::vector<unsigned int> subscripts; in gatherTransformFeedbackVaryings() local
|
| D | VaryingPacking.cpp | 783 std::vector<unsigned int> subscripts; in collectTFVarying() local
|
| /external/angle/src/libANGLE/renderer/d3d/ |
| D | ProgramExecutableD3D.cpp | 2114 std::vector<unsigned int> subscripts; in assignSamplerRegisters() local
2213 std::vector<unsigned int> subscripts; in assignImageRegisters() local
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_math_ops.py | 1349 def einsum(subscripts, *operands, **kwargs): # pylint: disable=missing-docstring argument
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | test_util.py | 3286 def _format_subscripts(self, subscripts, value, limit=10, indent=2): argument
|