Searched defs:indices_tensor (Results 1 – 6 of 6) sorted by relevance
527 const Tensor* indices_tensor; in Compute() local579 const Tensor* indices_tensor; in Compute() local628 const Tensor* indices_tensor; in Compute() local677 const Tensor* indices_tensor; in Compute() local
41 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
40 Status SparsifyWeights(const Tensor& tensor, Tensor* indices_tensor, in SparsifyWeights()350 Tensor indices_tensor; in SparsifyGatherInternal() local
346 Tensor indices_tensor(DT_INT64, shape_2d); in GrapplerItemFromMetaGraphDef() local