Home
last modified time | relevance | path

Searched refs:values_t (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsearchsorted_op.cc87 const Tensor& values_t = ctx->input(1); in Compute() local
90 OP_REQUIRES(ctx, sorted_inputs_t.dim_size(0) == values_t.dim_size(0), in Compute()
95 OP_REQUIRES(ctx, values_t.NumElements() < std::numeric_limits<int>::max(), in Compute()
100 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, values_t.shape(), &output_t)); in Compute()
113 const auto values = values_t.template flat<T>(); in Compute()
117 sorted_inputs_t.dim_size(1), values_t.dim_size(1), &output)); in Compute()
128 const Tensor& values_t = ctx->input(1); in Compute() local
131 OP_REQUIRES(ctx, sorted_inputs_t.dim_size(0) == values_t.dim_size(0), in Compute()
136 OP_REQUIRES(ctx, values_t.NumElements() < std::numeric_limits<int>::max(), in Compute()
141 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, values_t.shape(), &output_t)); in Compute()
[all …]
Dsparse_softmax_op.cc43 const Tensor *indices_t, *values_t, *shape_t; in Compute() local
45 OP_REQUIRES_OK(context, context->input("sp_values", &values_t)); in Compute()
54 TensorShapeUtils::IsVector(values_t->shape()) && in Compute()
59 values_t->shape().DebugString(), " and ", in Compute()
75 tensor::DeepCopy(*indices_t), tensor::DeepCopy(*values_t), in Compute()
Dsparse_dense_binary_op_shared.cc62 const Tensor *indices_t, *values_t, *shape_t, *dense_t; in Compute() local
64 OP_REQUIRES_OK(ctx, ctx->input("sp_values", &values_t)); in Compute()
74 TensorShapeUtils::IsVector(values_t->shape()) && in Compute()
79 values_t->shape().DebugString(), " and ", in Compute()
160 values_t->flat<T>().binaryExpr(dense_gathered_flat, in Compute()
Dsparse_fill_empty_rows_op.cc44 const Tensor* values_t; in Compute() local
48 OP_REQUIRES_OK(context, context->input("values", &values_t)); in Compute()
60 OP_REQUIRES(context, TensorShapeUtils::IsVector(values_t->shape()), in Compute()
62 values_t->shape().DebugString())); in Compute()
72 const auto values = values_t->vec<T>(); in Compute()
Dsparse_reduce_op.cc161 const Tensor *indices_t, *values_t, *shape_t, *reduction_axes_t; in Compute() local
163 OP_REQUIRES_OK(ctx, ctx->input("input_values", &values_t)); in Compute()
177 tensor::DeepCopy(*indices_t), tensor::DeepCopy(*values_t), in Compute()
257 const Tensor *indices_t, *values_t, *shape_t, *reduction_axes_t; in Compute() local
259 OP_REQUIRES_OK(ctx, ctx->input("input_values", &values_t)); in Compute()
267 tensor::DeepCopy(*values_t), in Compute()
Dreader_ops.cc143 auto values_t = values->vec<string>(); in ComputeWithReader() local
146 values_t(i) = std::move(values_vec[i]); in ComputeWithReader()
Dctc_decoder_ops.cc145 auto values_t = p_values->vec<int64>(); in StoreAllDecodedSequences() local
155 std::copy_n(p_batch.begin(), num_decoded, &values_t(offset)); in StoreAllDecodedSequences()
/external/tensorflow/tensorflow/python/ops/
Dsparse_ops_test.py76 values_t = constant_op.constant(values)
78 indices=indices_t, values=values_t, dense_shape=dense_shape_t)