Searched refs:a_values_tensor (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_1/ |
| H A D | multiplex_1_kernel.cc | 41 const auto& a_values_tensor = ctx->input(1); in Compute() local
48 OP_REQUIRES(ctx, a_values_tensor.shape() == b_values_tensor.shape(), in Compute()
52 a_values_tensor.shape().DebugString(), " b_values shape: ", in Compute()
55 ctx, a_values_tensor.shape() == cond_tensor.shape(), in Compute()
58 a_values_tensor.shape().DebugString(), in Compute()
61 const auto a_values = a_values_tensor.flat<T>(); in Compute()
68 ctx, ctx->allocate_output(0, a_values_tensor.shape(), &output_tensor)); in Compute()
70 const int64_t N = a_values_tensor.NumElements(); in Compute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_2/ |
| H A D | multiplex_2_kernel.h | 54 const auto& a_values_tensor = ctx->input(1); in Compute() local
61 OP_REQUIRES(ctx, a_values_tensor.shape() == b_values_tensor.shape(), in Compute()
65 a_values_tensor.shape().DebugString(), in Compute()
67 OP_REQUIRES(ctx, a_values_tensor.shape() == cond_tensor.shape(), in Compute()
71 a_values_tensor.shape().DebugString(), in Compute()
73 OP_REQUIRES(ctx, a_values_tensor.NumElements() > 0, in Compute()
77 const auto a_values = a_values_tensor.flat<T>(); in Compute()
84 ctx, ctx->allocate_output(0, a_values_tensor.shape(), &output_tensor)); in Compute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_3/ |
| H A D | multiplex_3_kernel.cc | 44 const auto& a_values_tensor = ctx->input(4); in Compute() local
52 OP_REQUIRES_OK(ctx, ValidateSparseTensor(a_indices_tensor, a_values_tensor, in Compute()
78 const auto a_values = a_values_tensor.flat<T>(); in Compute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_4/ |
| H A D | multiplex_4_kernel.cc | 99 const auto& a_values_tensor = ctx->input(num_cond_a_ + list_index); in Compute() local
101 const auto a_values = a_values_tensor.template flat<T>(); in Compute()
|