Searched defs:a_indices (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_tensor_dense_matmul_op_test.cc | 26 Node* SparseTensorDenseMatMulNode(Graph* g, Node* a_indices, Node* a_values, in SparseTensorDenseMatMulNode()
46 Tensor a_indices(DT_INT64, TensorShape({nnz, 2})); in SparseTensorDenseMatmul() local
|
| D | sparse_tensor_dense_matmul_op.cc | 43 const Tensor* a_indices; in Compute() local
256 typename TTypes<Tindices>::ConstMatrix a_indices, in SparseTensorDenseMatMulImpl()
|
| D | sparse_add_grad_op.cc | 32 const Tensor *backprop_val_grad, *a_indices, *b_indices, *sum_indices; in Compute() local
|
| D | sparse_add_op.cc | 33 const Tensor *a_indices, *b_indices, *a_values_t, *b_values_t, *a_shape, in Compute() local
|
| D | sparse_tensor_dense_add_op.cc | 36 Status ValidateInputs(const Tensor *a_indices, const Tensor *a_values, in ValidateInputs()
|
| D | sparse_tensor_dense_matmul_op_gpu.cu.cc | 48 const Tindices* __restrict__ a_indices, const T* __restrict__ a_values, in SparseTensorDenseMatMulKernel()
|
| /external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/ |
| D | sparse_add_op_test.py | 209 a_indices, argument
|
| /external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_3/ |
| D | multiplex_3_kernel.cc | 77 const auto a_indices = a_indices_tensor.matrix<int64_t>(); in Compute() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_ops.cc | 55 ShapeHandle a_indices; in __anonc55240e60202() local
|