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
|