Searched refs:values_tensor (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/testutil/ |
| D | batch_features_testutil.cc | 70 auto values_tensor = Tensor(tensorflow::DT_FLOAT, {sparse_values_size}); in RandomlyInitializeBatchFeatures() local
71 tensorflow::test::FillValues<float>(&values_tensor, values); in RandomlyInitializeBatchFeatures()
72 sparse_float_feature_values_list.push_back(values_tensor); in RandomlyInitializeBatchFeatures()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparsemask_op_test.py | 40 values_tensor = ops.convert_to_tensor(values)
44 t = ops.IndexedSlices(values_tensor, indices_tensor)
|
| /external/tensorflow/tensorflow/contrib/libsvm/kernels/ |
| D | decode_libsvm_op.cc | 117 Tensor* values_tensor; in Compute() local
121 &values_tensor)); in Compute()
122 auto values = values_tensor->vec<T>(); in Compute()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | sparsify_gather.cc | 42 Tensor* values_tensor) { in SparsifyWeights() argument
74 *values_tensor = in SparsifyWeights()
77 values_tensor->flat<float>().data()); in SparsifyWeights()
352 Tensor values_tensor; in SparsifyGatherInternal() local
354 SparsifyWeights(weight, &indices_tensor, &values_tensor)); in SparsifyGatherInternal()
365 CreateConstNode(values_tensor, StrCat(weights_node.name(), "/values"), in SparsifyGatherInternal()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | histogram_op.cc | 80 const Tensor& values_tensor = ctx->input(0); in Compute() local
92 const auto values = values_tensor.flat<T>(); in Compute()
|
| D | map_stage_op.cc | 523 OpInputList values_tensor; in Compute() local
527 OP_REQUIRES_OK(ctx, ctx->input_list("values", &values_tensor)); in Compute()
533 for (std::size_t i = 0; i < values_tensor.size(); ++i) { in Compute()
534 tuple.push_back(values_tensor[i]); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | quantile_ops.cc | 468 const Tensor& values_tensor = float_features_list[feature_idx]; in Compute() local
469 const int64 num_values = values_tensor.dim_size(0); in Compute()
479 auto flat_values = values_tensor.flat<float>(); in Compute()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | quantile_ops.cc | 112 const Tensor& values_tensor = values_list[feature_index]; in QuantizeFeatures() local
113 const int64 num_values = values_tensor.dim_size(0); in QuantizeFeatures()
125 auto flat_values = values_tensor.flat<float>(); in QuantizeFeatures()
|