Home
last modified time | relevance | path

Searched defs:indices_t (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_reduce_op.cc163 const Tensor *indices_t, *values_t, *shape_t, *reduction_axes_t; in Compute() local
275 const Tensor *indices_t, *values_t, *shape_t, *reduction_axes_t; in Compute() local
H A Done_hot_op_test.cc33 auto indices_t = indices.flat<int32>(); in OneHot() local
H A Dsparse_softmax_op.cc44 const Tensor *indices_t, *values_t, *shape_t; in Compute() local
H A Dsparse_dense_binary_op_shared.cc62 const Tensor *indices_t, *values_t, *shape_t, *dense_t; in Compute() local
H A Dtensor_array_ops.cc627 const auto indices_t = tensor_indices->vec<int32>(); in Compute() local
1074 const auto indices_t = tensor_indices->vec<int32>(); in Compute() local
H A Done_hot_op.cc110 auto indices_t = in Compute() local
H A Dctc_decoder_ops.cc149 auto indices_t = p_indices->matrix<int64_t>(); in StoreAllDecodedSequences() local
H A Dsparse_fill_empty_rows_op.cc217 const Tensor& indices_t = context->input(kIndicesInput); in SparseFillEmptyRowsOpImpl() local
H A Dragged_cross_op.cc127 SparseFeatureReader(const Tensor& indices_t, const Tensor& values_t, in SparseFeatureReader()
H A Dexample_parsing_ops.cc1048 auto indices_t = sp_indices_d->matrix<int64_t>(); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/
H A Dcsr_sparse_matrix_to_dense_op.cc151 Tensor indices_t; in Compute() local
H A Dcsr_sparse_matrix_to_sparse_tensor_op.cc157 Tensor* indices_t; in Compute() local
H A Dsparse_tensor_to_csr_sparse_matrix_op.cc141 const Tensor& indices_t = c->input(0); in ComputeAsync() local
H A Ddense_to_csr_sparse_matrix_op.cc160 const Tensor& indices_t = c->input(1); in ComputeAsync() local
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Ddata_flow_ops.cc99 const Tensor* indices_t = c->input_tensor(i); in DynamicStitchShapeFunction() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dconstant_folding_test.cc3224 Tensor indices_t(DT_INT32, TensorShape({2})); in TEST_F() local