Home
last modified time | relevance | path

Searched refs:indices_tensor (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dunravel_index_op.cc41 const Tensor& indices_tensor = ctx->input(0); in Compute() local
43 TensorShapeUtils::IsVector(indices_tensor.shape()) || in Compute()
44 TensorShapeUtils::IsScalar(indices_tensor.shape()), in Compute()
47 indices_tensor.shape().DebugString(), "\"")); in Compute()
82 if (TensorShapeUtils::IsScalar(indices_tensor.shape())) { in Compute()
89 output = output.constant(indices_tensor.scalar<Tidx>()()); in Compute()
95 indices_tensor.NumElements()}), in Compute()
101 Eigen::array<Eigen::Index, 2> bcast({1, indices_tensor.NumElements()}); in Compute()
103 {1, indices_tensor.NumElements()}}; in Compute()
107 output = indices_tensor.vec<Tidx>() in Compute()
Dmap_stage_op.cc522 const Tensor* indices_tensor; in Compute() local
526 OP_REQUIRES_OK(ctx, ctx->input("indices", &indices_tensor)); in Compute()
538 OP_REQUIRES_OK(ctx, map->put(&key, indices_tensor, &tuple)); in Compute()
584 const Tensor* indices_tensor; in Compute() local
587 OP_REQUIRES_OK(ctx, ctx->input("indices", &indices_tensor)); in Compute()
588 OP_REQUIRES_OK(ctx, map->pop(key_tensor, indices_tensor, &tuple)); in Compute()
591 ctx, tuple.size() == indices_tensor->NumElements(), in Compute()
593 " vs. ", indices_tensor->NumElements())); in Compute()
645 const Tensor* indices_tensor; in Compute() local
648 OP_REQUIRES_OK(ctx, ctx->input("indices", &indices_tensor)); in Compute()
[all …]
/external/tensorflow/tensorflow/core/kernels/fuzzing/
Dscatter_nd_fuzz.cc90 Tensor indices_tensor(tensorflow::DT_INT32, TensorShape(indices_dims)); in FuzzImpl() local
93 auto flat_indices = indices_tensor.flat<int32>(); in FuzzImpl()
122 RunInputs({{"indices", indices_tensor}, in FuzzImpl()
/external/tensorflow/tensorflow/python/kernel_tests/
Dsparsemask_op_test.py41 indices_tensor = ops.convert_to_tensor(indices)
44 t = ops.IndexedSlices(values_tensor, indices_tensor)
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dpooling.cc188 TensorCodeGenerator indices_tensor( in GetMaxPoolingKernelCode() local
199 c += indices_tensor.GetDeclaration(AccessType::WRITE) + ",\n"; in GetMaxPoolingKernelCode()
265 c += " " + indices_tensor.WriteWHS("indexes", "X", "Y", "Z"); in GetMaxPoolingKernelCode()
284 TensorCodeGenerator indices_tensor( in GetMaxPooling3DKernelCode() local
296 c += indices_tensor.GetDeclaration(AccessType::WRITE) + ",\n"; in GetMaxPooling3DKernelCode()
370 c += " " + indices_tensor.WriteWHDS("indexes", "X", "Y", "Z", "S"); in GetMaxPooling3DKernelCode()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dsparsify_gather.cc40 Status SparsifyWeights(const Tensor& tensor, Tensor* indices_tensor, in SparsifyWeights() argument
68 *indices_tensor = Tensor(DataTypeToEnum<int64>::value, in SparsifyWeights()
71 indices_tensor->flat<int64>().data()); in SparsifyWeights()
350 Tensor indices_tensor; in SparsifyGatherInternal() local
353 SparsifyWeights(weight, &indices_tensor, &values_tensor)); in SparsifyGatherInternal()
358 CreateConstNode(indices_tensor, in SparsifyGatherInternal()
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_builder.cc345 Tensor indices_tensor(DT_INT64, shape_2d); in GrapplerItemFromMetaGraphDef() local
346 InitializeTensor(input.dtype(), &indices_tensor); in GrapplerItemFromMetaGraphDef()
349 indices_tensor); in GrapplerItemFromMetaGraphDef()