| /external/tensorflow/tensorflow/core/kernels/ |
| sparse_slice_grad_op.cc | 30 const Tensor *backprop_val_grad, *input_indices, *output_indices, *input_start; variable
34 OP_REQUIRES_OK(ctx, ctx->input("output_indices", &output_indices));
38 TensorShapeUtils::IsMatrix(output_indices->shape()),
43 output_indices->shape().DebugString()));
51 input_indices->dim_size(1) == output_indices->dim_size(1),
54 output_indices->dim_size(1)));
56 ctx, output_indices->dim_size(0) <= input_indices->dim_size(0),
57 errors::InvalidArgument("# rows of output_indices should be not greater "
59 output_indices->dim_size(0), " and "
[all...] |
| deserialize_sparse_variant_op.cc | 59 const Tensor* output_indices; variable
69 output_shape->NumElements(), &output_indices,
71 context->set_output(0, *output_indices);
152 Tensor* output_indices; variable
159 &output_indices));
169 int64* output_indices_data = output_indices->matrix<int64>().data();
198 // Build the submatrix of `output_indices` for the i^th sparse tensor
201 // Each row of `output_indices` comprises `input_dims_to_stack` indices
304 int index, int expected_rank, const Tensor** output_indices,
307 *output_indices = serialized_indices.get<Tensor>()
[all...] |
| deserialize_sparse_string_op.cc | 80 Tensor output_indices; variable
87 &output_indices, &output_values, &output_shape));
88 context->set_output(0, output_indices);
103 Tensor output_indices; variable
109 serialized_sparse_t(i, 2), dtype_, i, &output_indices,
111 int64 num_entries = output_indices.dim_size(0);
112 int rank = output_indices.dim_size(1);
117 const auto& output_indices_t = output_indices.matrix<int64>();
229 Tensor* output_indices, Tensor* output_values, Tensor* output_shape) {
231 TF_RETURN_IF_ERROR(this->Deserialize(serialized_indices, output_indices));
[all...] |
| sparse_tensors_map_ops.cc | 295 Tensor output_indices = Tensor(DT_INT64, {num_entries, rank - 1}); variable
298 auto output_indices_t = output_indices.matrix<int64>();
310 SparseTensor::Create(output_indices, output_values,
387 const Tensor& output_indices = st.indices(); variable
391 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(output_indices.shape()),
395 output_indices.shape().DebugString()));
408 int64 num_entries = output_indices.dim_size(0);
415 int rank = output_indices.dim_size(1);
427 DT_INT64, TensorShape({num_entries, 1 + output_indices.dim_size(1)}));
429 const auto& output_indices_t = output_indices.matrix<int64>()
[all...] |
| sparse_fill_empty_rows_op.cc | 99 OP_REQUIRES_OK(context, context->allocate_output("output_indices",
142 OP_REQUIRES_OK(context, context->allocate_output("output_indices",
145 auto output_indices = output_indices_t->matrix<int64>(); variable
146 output_indices.device(d) = output_indices.constant(0);
168 std::copy_n(&indices(i, 0), rank, &output_indices(output_i, 0));
182 output_indices(starting_index, 0) = row;
|
| serialize_sparse_op.cc | 240 Tensor output_indices = Tensor(DT_INT64, {num_entries, rank - 1}); variable
243 auto output_indices_t = output_indices.matrix<int64>();
254 context, this->Serialize(output_indices, &serialized_sparse_t(b, 0)));
|
| decode_proto_op.cc | 681 std::vector<int> output_indices; local
682 output_indices.reserve(field_names.size());
684 output_indices.push_back(i);
686 std::sort(output_indices.begin(), output_indices.end(),
693 const int output_index = output_indices[i];
[all...] |
| non_max_suppression_op.cc | 218 Tensor* output_indices = nullptr; local
221 context->allocate_output(0, output_shape, &output_indices));
222 TTypes<int, 1>::Tensor output_indices_data = output_indices->tensor<int, 1>();
|
| /external/libtextclassifier/utils/tflite/ |
| dist_diversification.cc | 83 TfLiteTensor& output_indices = local
86 return context->ResizeTensor(context, &output_indices,
97 TfLiteTensor& output_indices = local
100 tflite::SetTensorToDynamic(&output_indices);
110 TfLiteTensor& output_indices = local
113 if (tflite::IsDynamicTensor(&output_indices)) {
133 std::copy(indices.begin(), indices.end(), output_indices.data.i32);
134 std::fill_n(output_indices.data.i32 + indices.size(),
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| reverse_test.cc | 93 std::vector<int64> output_indices(spec.input_dims.size());
96 output_indices[i] = indices[i];
100 output_indices[dim] = (spec.input_dims[dim] - 1) - indices[dim];
102 expected.Set<float>(output_indices, value);
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| resolve_constant_binary.cc | 126 const std::vector<int> output_indices = ReverseOffset(output_shape, k); local
128 modulo_indices[i] = output_indices[i] % small_sizes[i];
130 PairwiseVectorSelect(input0_larger, output_indices, modulo_indices,
163 output_data[Offset(output_shape, output_indices)] = outval;
|
| resolve_constant_unary.cc | 64 std::vector<int> output_indices(input_shape.dimensions_count());
70 output_indices[i] = input_indices[i] * reduction_mask[i];
72 int output_offset = Offset(output_shape, output_indices);
73 if (input_indices == output_indices) {
|
| /external/tensorflow/tensorflow/python/ops/ |
| sparse_ops.py | [all...] |
| sparse_grad.py | 139 output_indices = op.outputs[0]
142 backprop_val_grad, input_indices, input_start, output_indices)
|
| /external/tensorflow/tensorflow/core/ops/ |
| sparse_ops.cc | 211 .Output("output_indices: int64")
257 .Output("output_indices: int64")
280 .Output("output_indices: num_split * int64")
287 ShapeHandle output_indices =
296 c->set_output(out_idx++, output_indices);
308 .Input("output_indices: int64")
324 .Output("output_indices: int64")
330 ShapeHandle output_indices =
335 c->set_output(0, output_indices);
345 .Output("output_indices: int64"
[all...] |
| /external/tensorflow/tensorflow/python/eager/ |
| ops_test.py | 143 output_indices, output_values, output_shape = result
144 self.assertEqual(2, len(output_indices))
147 self.assertEqual(output_indices, result.output_indices)
150 self.assertAllEqual([[0, 2], [1, 0], [1, 1]], output_indices[0])
151 self.assertAllEqual([[0, 0], [0, 1]], output_indices[1])
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| graph_jni.cc | 216 jintArray output_indices = env->NewIntArray(noutputs); local
224 env->GetIntArrayElements(output_indices, nullptr);
239 env->ReleaseIntArrayElements(output_indices, output_indices_elems, 0);
245 input_handles, input_indices, output_handles, output_indices);
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| sparse_tensor.h | 575 std::vector<Tensor> output_indices; local
578 output_indices.reserve(num_split);
614 output_indices.emplace_back(DT_INT64,
619 output_indices_t.emplace_back(output_indices[i].matrix<int64>());
645 Create(output_indices[i], output_values[i], output_shapes[i], &tensor);
696 Tensor output_indices(DT_INT64, TensorShape({count, dims}));
699 auto output_indices_t = output_indices.matrix<int64>();
726 return SparseTensor(output_indices, output_values, output_shape);
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| literal.h | [all...] |
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
