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()
60 const Tidx* indices = indices_tensor.flat<Tidx>().data(); in Compute()
61 int64 size = indices_tensor.NumElements(); in Compute()
92 if (TensorShapeUtils::IsScalar(indices_tensor.shape())) { in Compute()
99 output = output.constant(indices_tensor.scalar<Tidx>()()); in Compute()
105 indices_tensor.NumElements()}), in Compute()
113 {1, static_cast<Eigen::Index>(indices_tensor.NumElements())}); in Compute()
[all …]
Dmap_stage_op.cc524 const Tensor* indices_tensor; in Compute() local
528 OP_REQUIRES_OK(ctx, ctx->input("indices", &indices_tensor)); in Compute()
540 OP_REQUIRES_OK(ctx, map->put(&key, indices_tensor, &tuple)); in Compute()
574 const Tensor* indices_tensor; in Compute() local
577 OP_REQUIRES_OK(ctx, ctx->input("indices", &indices_tensor)); in Compute()
578 OP_REQUIRES_OK(ctx, map->pop(key_tensor, indices_tensor, &tuple)); in Compute()
581 ctx, tuple.size() == indices_tensor->NumElements(), in Compute()
583 " vs. ", indices_tensor->NumElements())); in Compute()
623 const Tensor* indices_tensor; in Compute() local
626 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/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/python/ops/ragged/
Dragged_gather_op_test.py383 for indices_tensor in indices_tensors:
385 params_tensor, indices_tensor, axis=axis, batch_dims=batch_dims)
391 0), getattr(indices_tensor, 'ragged_rank', 0)))
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_builder.cc346 Tensor indices_tensor(DT_INT64, shape_2d); in GrapplerItemFromMetaGraphDef() local
347 InitializeTensor(input.dtype(), &indices_tensor); in GrapplerItemFromMetaGraphDef()
350 indices_tensor); in GrapplerItemFromMetaGraphDef()