| /external/tensorflow/tensorflow/core/kernels/ |
| D | unravel_index_op.cc | 41 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()
|
| D | map_stage_op.cc | 522 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/contrib/tensor_forest/kernels/ |
| D | scatter_add_ndim_op.cc | 35 const Tensor& indices_tensor = context->input(1); in Compute() local
38 if (indices_tensor.shape().dim_size(0) > 0) { in Compute()
39 OP_REQUIRES(context, indices_tensor.shape().dims() == 2, in Compute()
44 indices_tensor.shape().dim_size(1) + delta_dims == in Compute()
51 indices_tensor.shape().dim_size(0) == in Compute()
61 if (!CheckTensorBounds(context, indices_tensor)) return; in Compute()
66 const auto indices = indices_tensor.tensor<int32, 2>(); in Compute()
70 static_cast<int32>(indices_tensor.shape().dim_size(1)); in Compute()
93 for (int32 i = 0; i < indices_tensor.shape().dim_size(0); i++) { in Compute()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/testutil/ |
| D | batch_features_testutil.cc | 62 auto indices_tensor = Tensor(tensorflow::DT_INT64, {sparse_values_size, 2}); in RandomlyInitializeBatchFeatures() local
63 tensorflow::test::FillValues<int64>(&indices_tensor, indices_vector); in RandomlyInitializeBatchFeatures()
64 sparse_float_feature_indices_list.push_back(indices_tensor); in RandomlyInitializeBatchFeatures()
|
| /external/tensorflow/tensorflow/core/kernels/fuzzing/ |
| D | scatter_nd_fuzz.cc | 90 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/ |
| D | sparsemask_op_test.py | 41 indices_tensor = ops.convert_to_tensor(indices)
44 t = ops.IndexedSlices(values_tensor, indices_tensor)
|
| /external/tensorflow/tensorflow/contrib/libsvm/kernels/ |
| D | decode_libsvm_op.cc | 93 Tensor* indices_tensor; in Compute() local
98 &indices_tensor)); in Compute()
99 auto indices = indices_tensor->matrix<int64>(); in Compute()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | sparsify_gather.cc | 41 Status SparsifyWeights(const Tensor& tensor, Tensor* indices_tensor, in SparsifyWeights() argument
69 *indices_tensor = Tensor(DataTypeToEnum<int64>::value, in SparsifyWeights()
72 indices_tensor->flat<int64>().data()); in SparsifyWeights()
351 Tensor indices_tensor; in SparsifyGatherInternal() local
354 SparsifyWeights(weight, &indices_tensor, &values_tensor)); in SparsifyGatherInternal()
359 CreateConstNode(indices_tensor, in SparsifyGatherInternal()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item_builder.cc | 246 Tensor indices_tensor(DT_INT64, shape_2d); in GrapplerItemFromMetaGraphDef() local
247 InitializeTensor(input.dtype(), &indices_tensor); in GrapplerItemFromMetaGraphDef()
250 indices_tensor); in GrapplerItemFromMetaGraphDef()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 3729 TRT_TensorOrWeights indices_tensor = inputs.at(1); in ConvertGather() local
3730 if (indices_tensor.batch_size() != 1) { in ConvertGather()
3737 indices_tensor.GetTrtDims().nbDims + 1; in ConvertGather()
3752 *const_cast<nvinfer1::ITensor*>(indices_tensor.tensor()), trt_axis); in ConvertGather()
|