Searched defs:indices_tensor (Results 1 – 5 of 5) sorted by relevance
524 const Tensor* indices_tensor; in Compute() local574 const Tensor* indices_tensor; in Compute() local623 const Tensor* indices_tensor; in Compute() local672 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
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