Searched refs:x_input (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | listdiff_op.cc | 73 std::vector<int64> x_input, y_input; in ListDiff() local
74 TF_RETURN_IF_ERROR(context->ConstantInputAsIntVector(0, &x_input)); in ListDiff()
85 auto x_size = x_input.size(); in ListDiff()
87 if (y_input_set.count(x_input[i]) > 0) { in ListDiff()
90 val_output.push_back(x_input[i]); in ListDiff()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 93 void operator()(OpKernelContext* context, const Tensor& x_input, in operator ()()
122 const int64 in_batch = GetTensorDim(x_input, tensor_format, 'N'); in operator ()()
123 const int64 in_rows = GetTensorDim(x_input, tensor_format, 'H'); in operator ()()
124 const int64 in_cols = GetTensorDim(x_input, tensor_format, 'W'); in operator ()()
125 const int64 in_depths = GetTensorDim(x_input, tensor_format, 'C'); in operator ()()
140 x_input, perm, &transformed_x)); in operator ()()
142 transformed_x = x_input; in operator ()()
231 void operator()(OpKernelContext* context, const Tensor& x_input, in operator ()()
260 const int64 in_batch = GetTensorDim(x_input, tensor_format, 'N'); in operator ()()
261 const int64 in_rows = GetTensorDim(x_input, tensor_format, 'H'); in operator ()()
[all …]
|
| D | fused_batch_norm_op.h | 68 const Tensor& x_input, const Tensor& scale_input, in operator()
|
| D | cwise_ops_common.h | 231 const Tensor& x_input = context->input(0); in Compute() local
234 context, x_input.shape() == y_input.shape(), in Compute()
236 "x shape: ", x_input.shape().DebugString(), in Compute()
240 context->allocate_output(0, x_input.shape(), &z_output)); in Compute()
242 typename TTypes<T>::ConstFlat x(x_input.flat<T>()); in Compute()
|
| D | fused_batch_norm_op.cu.cc | 37 const Tensor& x_input, const Tensor& scale_input, in operator ()()
44 typename TTypes<T, 4>::ConstTensor input(x_input.tensor<T, 4>()); in operator ()()
|
| /external/rust/crates/rusqlite/src/ |
| D | session.rs | 221 Some(x_input), in invert_strm()
241 Some(x_input), in concat_strm()
243 Some(x_input), in concat_strm()
314 Some(x_input), in start_strm()
514 Some(x_input), in add_stream()
606 Some(x_input), in apply_strm()
615 Some(x_input), in apply_strm()
709 unsafe extern "C" fn x_input(p_in: *mut c_void, data: *mut c_void, len: *mut c_int) -> c_int { in x_input() function
|