Searched defs:indices_tensor (Results 1 – 9 of 9) sorted by relevance
35 const Tensor& indices_tensor = context->input(1); in Compute() local
41 const Tensor& indices_tensor = ctx->input(0); in Compute() local
522 const Tensor* indices_tensor; in Compute() local584 const Tensor* indices_tensor; in Compute() local645 const Tensor* indices_tensor; in Compute() local704 const Tensor* indices_tensor; in Compute() local
90 Tensor indices_tensor(tensorflow::DT_INT32, TensorShape(indices_dims)); in FuzzImpl() local
62 auto indices_tensor = Tensor(tensorflow::DT_INT64, {sparse_values_size, 2}); in RandomlyInitializeBatchFeatures() local
93 Tensor* indices_tensor; in Compute() local
41 Status SparsifyWeights(const Tensor& tensor, Tensor* indices_tensor, in SparsifyWeights()351 Tensor indices_tensor; in SparsifyGatherInternal() local
246 Tensor indices_tensor(DT_INT64, shape_2d); in GrapplerItemFromMetaGraphDef() local
3729 TRT_TensorOrWeights indices_tensor = inputs.at(1); in ConvertGather() local