| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | matrix_triangular_solve_op_test.py | 32 def _verifySolveAllWays(self, x, y, dtypes, batch_dims=None): argument
45 def _verifySolveAllWaysReal(self, x, y, batch_dims=None): argument
48 def _verifySolveAllWaysComplex(self, x, y, batch_dims=None): argument
56 batch_dims=None, argument
|
| D | gather_op_test.py | 335 def testBatchDims(self, params, indices, batch_dims, expected=None, argument
408 batch_dims, axis, output_shape): argument
434 def _batchNumpyGather(self, params, indices, axis, batch_dims): argument
|
| D | matrix_solve_op_test.py | 39 def _verifySolve(self, x, y, batch_dims=None): argument
|
| D | matrix_exponential_op_test.py | 218 def _TestRandomSmall(dtype, batch_dims, size): argument
|
| D | resource_variable_ops_test.py | 1160 def testGatherWithBatchDims(self, params, indices, batch_dims, expected): argument
1206 batch_dims, output_shape): argument
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_gather_ops.py | 36 def gather(params, indices, validate_indices=None, axis=0, batch_dims=0, argument
|
| D | ragged_dispatch.py | 407 axis=0, batch_dims=0): argument
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | qr.cc | 75 Status House(XlaOp x, XlaOp k, absl::Span<const int64> batch_dims, in House()
168 std::vector<int64> batch_dims(num_batch_dims); in QRBlock() local
252 absl::Span<const int64> batch_dims, in ComputeWYRepresentation()
345 std::vector<int64> batch_dims(num_batch_dims); in QRDecomposition() local
|
| D | svd.cc | 124 std::vector<int64> batch_dims(num_batch_dims); in HouseRow() local
190 std::vector<int64> batch_dims(num_batch_dims); in HouseCol() local
264 std::vector<int64> batch_dims(num_batch_dims); in HouseHolderBidiagonalization() local
469 std::vector<int64> batch_dims(num_batch_dims); in OneSidedJacobiUpdate() local
838 std::vector<int64> batch_dims(num_batch_dims); in SVD() local
|
| D | self_adjoint_eig.cc | 120 const std::vector<int64> batch_dims(w_shape.dimensions().begin(), in Update() local
430 std::vector<int64> batch_dims(num_batch_dims); in SelfAdjointEig() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | resource_variable_ops.cc | 270 int32 batch_dims; in __anon3f9cfa090402() local
|
| D | math_ops.cc | 144 ShapeHandle batch_dims; in __anonb22bfa860202() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | softmax_op.cc | 47 std::vector<int64> batch_dims(logits_shape.dims() - 1); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | indexed_array_analysis.cc | 977 absl::Span<const int64> batch_dims) { in GetOnlyNonContractingNonBatchDim()
1002 absl::Span<const int64> batch_dims) { in CanFoldDotIntoIndexedArray()
|
| D | dynamic_dimension_inference.cc | 210 std::unordered_set<int64> batch_dims( in HandleDot() local
|
| D | shape_inference.cc | 563 absl::Span<const int64> batch_dims) -> bool { in ValidateDotDimensionNumbers()
588 absl::Span<const int64> batch_dims) -> bool { in ValidateDotDimensionNumbers()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 3293 batch_dims=0): argument
3381 batch_dims=0, name=None): argument
3405 def _batch_gather(params, indices, batch_dims, axis=None): argument
|