Searched defs:indices_tensor (Results 1 – 10 of 10) 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
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
40 Status SparsifyWeights(const Tensor& tensor, Tensor* indices_tensor, in SparsifyWeights()350 Tensor indices_tensor; in SparsifyGatherInternal() local
179 Status ValidateSparseTensor(const ::tensorflow::Tensor& indices_tensor, in ValidateSparseTensor()
347 Tensor indices_tensor(DT_INT64, shape_2d); in GrapplerItemFromMetaGraphDef() local
434 std::unique_ptr<parallel_device::ParallelTensor> indices_tensor, in Wrap()
4400 ITensorProxyPtr indices_tensor = populate_tensor(indices_input); in ConvertGather() local