Searched defs:a_indices (Results 1 – 7 of 7) 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_add_grad_op.cc | 32 const Tensor *backprop_val_grad, *a_indices, *b_indices, *sum_indices; in Compute() local
|
| D | sparse_tensor_dense_matmul_op_gpu.cu.cc | 32 const Tindices* __restrict__ a_indices, const T* __restrict__ a_values, in SparseTensorDenseMatMulKernel()
|
| D | sparse_tensor_dense_add_op.cc | 35 Status ValidateInputs(const Tensor *a_indices, const Tensor *a_values, in ValidateInputs()
|
| D | sparse_add_op.cc | 32 const Tensor *a_indices, *b_indices, *a_values_t, *b_values_t, *a_shape, in Compute() local
|
| D | sparse_tensor_dense_matmul_op.cc | 43 const Tensor* a_indices; in Compute() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_ops.cc | 53 ShapeHandle a_indices; in __anon36a8fc880202() local
|