Searched refs:b_values_tensor (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_4/ |
| H A D | multiplex_4_kernel.cc | 57 const auto& b_values_tensor = ctx->input(2 * num_cond_a_); in Compute() local
67 ctx, a_values_tensor_i.shape() == b_values_tensor.shape(), in Compute()
73 " b_values shape: ", b_values_tensor.shape().DebugString())); in Compute()
75 ctx, cond_tensor_i.shape() == b_values_tensor.shape(), in Compute()
87 ctx, ctx->allocate_output(0, b_values_tensor.shape(), &output_tensor)); in Compute()
90 const auto b_values = b_values_tensor.template flat<T>(); in Compute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_1/ |
| H A D | multiplex_1_kernel.cc | 42 const auto& b_values_tensor = ctx->input(2); in Compute() local
48 OP_REQUIRES(ctx, a_values_tensor.shape() == b_values_tensor.shape(), in Compute()
53 b_values_tensor.shape().DebugString())); in Compute()
62 const auto b_values = b_values_tensor.flat<T>(); in Compute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_2/ |
| H A D | multiplex_2_kernel.h | 55 const auto& b_values_tensor = ctx->input(2); in Compute() local
61 OP_REQUIRES(ctx, a_values_tensor.shape() == b_values_tensor.shape(), in Compute()
66 " b shape: ", b_values_tensor.shape().DebugString())); in Compute()
78 const auto b_values = b_values_tensor.flat<T>(); in Compute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_3/ |
| H A D | multiplex_3_kernel.cc | 47 const auto& b_values_tensor = ctx->input(7); in Compute() local
54 OP_REQUIRES_OK(ctx, ValidateSparseTensor(b_indices_tensor, b_values_tensor, in Compute()
81 const auto b_values = b_values_tensor.flat<T>(); in Compute()
|