| /external/tensorflow/tensorflow/core/kernels/ |
| sparse_slice_op.cc | 32 const Tensor& input_values = context->input(1); variable
41 OP_REQUIRES(context, TensorShapeUtils::IsVector(input_values.shape()),
72 input_indices, input_values,
|
| sparse_split_op.cc | 35 const Tensor& input_values = context->input(2); variable
42 OP_REQUIRES(context, TensorShapeUtils::IsVector(input_values.shape()),
69 input_indices, input_values,
|
| serialize_sparse_op.cc | 56 const Tensor* input_values; variable
60 OP_REQUIRES_OK(context, context->input("sparse_values", &input_values));
67 OP_REQUIRES(context, TensorShapeUtils::IsVector(input_values->shape()),
70 input_values->shape().DebugString()));
82 OP_REQUIRES_OK(context, Serialize(*input_values, &serialized_sparse_t(1)));
163 const Tensor* input_values; variable
166 OP_REQUIRES_OK(context, context->input("sparse_values", &input_values));
173 OP_REQUIRES(context, TensorShapeUtils::IsVector(input_values->shape()),
176 input_values->shape().DebugString()));
194 OP_REQUIRES_OK(context, SparseTensor::Create(*input_indices, *input_values,
[all...] |
| sparse_tensors_map_ops.cc | 170 const Tensor* input_values; variable
175 OP_REQUIRES_OK(context, context->input("sparse_values", &input_values));
184 OP_REQUIRES(context, TensorShapeUtils::IsVector(input_values->shape()),
187 input_values->shape().DebugString()));
200 OP_REQUIRES_OK(context, SparseTensor::Create(*input_indices, *input_values,
225 const Tensor* input_values; variable
230 OP_REQUIRES_OK(context, context->input("sparse_values", &input_values));
239 OP_REQUIRES(context, TensorShapeUtils::IsVector(input_values->shape()),
242 input_values->shape().DebugString()));
260 OP_REQUIRES_OK(context, SparseTensor::Create(*input_indices, *input_values,
[all...] |
| /external/tensorflow/tensorflow/lite/testing/ |
| generate_testspec.cc | 46 std::vector<string> input_values; local
47 input_values.resize(input_layer.size());
55 GenerateCsv<float>(shape, -0.5, 0.5, &input_values[i]);
58 GenerateCsv<uint8_t>(shape, 0, 255, &input_values[i]);
61 GenerateCsv<int32_t>(shape, -100, 100, &input_values[i]);
64 GenerateCsv<int64_t>(shape, -100, 100, &input_values[i]);
67 GenerateCsv<int>(shape, 0.01, 1.99, &input_values[i]);
72 input_values.clear();
73 return input_values;
76 return input_values;
115 std::vector<string> input_values = local
[all...] |
| /external/tensorflow/tensorflow/cc/framework/ |
| while_gradients_test.cc | 55 void Run(const std::vector<Input::Initializer>& input_values,
57 Run<T>(ClientSession(scope_), input_values, expected_grad_values); local
62 const std::vector<Input::Initializer>& input_values,
66 DCHECK_EQ(input_values.size(), inputs_.size());
69 feeds.emplace(inputs_[i], input_values[i]);
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| reduce_precision_test.cc | 212 std::vector<float> input_values; local
218 input_values.push_back(absl::bit_cast<float>(test_value[0]));
222 input_values.push_back(absl::bit_cast<float>(test_value[0] ^ sign_bit));
232 Literal a_literal = LiteralUtil::CreateR1<float>({input_values});
|
| dynamic_ops_test.cc | 126 Literal input_values = local
141 auto input = ConstantLiteral(&builder, input_values);
152 Literal input_values = local
171 auto input = ConstantLiteral(&builder, input_values);
187 Literal input_values = local
205 auto input = ConstantLiteral(&builder, input_values);
405 Literal input_values = local
424 auto input = ConstantLiteral(&builder, input_values);
436 Literal input_values = local
458 auto input = ConstantLiteral(&builder, input_values);
475 Literal input_values = local
[all...] |
| /external/tensorflow/tensorflow/contrib/factorization/kernels/ |
| wals_solver_ops.cc | 78 const Tensor& input_values = context->input(5); variable
97 OP_REQUIRES(context, TensorShapeUtils::IsVector(input_values.shape()),
98 InvalidArgument("Input input_values should be a vector"));
101 OP_REQUIRES(context, input_indices.dim_size(0) == input_values.dim_size(0),
102 InvalidArgument("Input input_values' length should match the "
129 const auto& input_values_vec = input_values.vec<float>();
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| propagate_fixed_sizes.cc | 1446 const auto& input_values = model->GetArray(op->inputs[0]); local
[all...] |
