Home
last modified time | relevance | path

Searched defs:indices_ (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_slice_grad_op_gpu.cu.cc70 const int64_t* __restrict__ indices_; member
85 const int64_t* indices_; member
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_reachability.h242 absl::flat_hash_map<uint64_t, int> indices_; variable
Dindexed_array_analysis.h204 Array* indices_; variable
/external/tensorflow/tensorflow/lite/kernels/
Dsparse_to_dense_test.cc85 int indices_; member in tflite::__anon83c1f70b0111::SparseToDenseOpModel
Dembedding_lookup_sparse_test.cc86 int indices_; member in tflite::__anon661b93d20111::EmbeddingLookupSparseOpModel
Dscatter_nd_test.cc67 int indices_; member in tflite::__anon1c86c5740111::ScatterNdOpModel
Done_hot_test.cc64 int indices_; member in tflite::__anon22e919f90111::OneHotOpModel
Dgather_nd_test.cc63 int indices_; member in tflite::__anon40db79d90111::GatherNdOpModel
/external/tensorflow/tensorflow/lite/kernels/perception/
Dmax_pool_with_argmax_test.cc66 int indices_; member in tflite::ops::custom::__anone125c3ce0111::MaxpoolingWithArgMaxOpModel
Dmax_unpooling_2d_test.cc65 int indices_; member in tflite::ops::custom::__anone465cf2a0111::MaxUnpoolingOpModel
/external/tensorflow/tensorflow/compiler/xla/python/
Dsharded_device_array.h324 pybind11::object indices_; variable
/external/webrtc/rtc_tools/frame_analyzer/
Dvideo_temporal_aligner.cc79 const std::vector<size_t> indices_; member in webrtc::test::__anon021d14220111::ReorderedVideo
/external/tensorflow/tensorflow/core/kernels/data/
Dsparse_tensor_slice_dataset_op_test.cc65 Tensor indices_; member in tensorflow::data::__anon4063b90b0111::SparseTensorSliceDatasetParams
/external/tensorflow/tensorflow/dtensor/cc/
Ddtensor_device_util.h510 std::unique_ptr<parallel_device::ParallelTensor> indices_; variable