Searched defs:indices_tensor (Results 1 – 12 of 12) sorted by relevance
530 const Tensor* indices_tensor; in Compute() local586 const Tensor* indices_tensor; in Compute() local633 const Tensor* indices_tensor; in Compute() local680 const Tensor* indices_tensor; in Compute() local
46 const Tensor& indices_tensor = ctx->input(0); in Compute() local
40 Status SparsifyWeights(const Tensor& tensor, Tensor* indices_tensor, in SparsifyWeights()350 Tensor indices_tensor; in SparsifyGatherInternal() local
90 Tensor indices_tensor(tensorflow::DT_INT32, TensorShape(indices_dims)); in FuzzImpl() local
85 TensorType indices_tensor = create_tensor<TensorType>(indices_shape, DataType::U32); in compute_target() local
159 ITensorProxyPtr indices_tensor = in Convert() local
179 Status ValidateSparseTensor(const ::tensorflow::Tensor& indices_tensor, in ValidateSparseTensor()
347 Tensor indices_tensor(DT_INT64, shape_2d); in GrapplerItemFromMetaGraphDef() local
902 auto& indices_tensor = elements.at(idx++).toTensor(); in rebuildSparseTensor() local
434 std::unique_ptr<parallel_device::ParallelTensor> indices_tensor, in Wrap()
1738 auto indices_tensor = at::empty({total_size + 1}, at::kInt); in sync_bucket_indices() local
4400 ITensorProxyPtr indices_tensor = populate_tensor(indices_input); in ConvertGather() local