| /external/tensorflow/tensorflow/core/kernels/ |
| D | searchsorted_op_gpu.cu.cc | 34 __global__ void UpperBoundKernel(const T* __restrict__ sorted_inputs, in UpperBoundKernel() argument
42 sorted_inputs + bid * sorted_inputs_size, sorted_inputs_size, value); in UpperBoundKernel()
47 __global__ void LowerBoundKernel(const T* __restrict__ sorted_inputs, in LowerBoundKernel() argument
55 sorted_inputs + bid * sorted_inputs_size, sorted_inputs_size, value); in LowerBoundKernel()
64 const typename TTypes<T, 1>::ConstTensor& sorted_inputs, in Compute()
77 config.thread_per_block, 0, device.stream(), sorted_inputs.data(), in Compute()
87 const typename TTypes<T, 1>::ConstTensor& sorted_inputs, in Compute()
100 config.thread_per_block, 0, device.stream(), sorted_inputs.data(), in Compute()
|
| D | searchsorted_op.cc | 36 const typename TTypes<T, 1>::ConstTensor& sorted_inputs, in Compute()
42 const T* sorted_inputs_ptr = sorted_inputs.data() + b * num_inputs; in Compute()
59 const typename TTypes<T, 1>::ConstTensor& sorted_inputs, in Compute()
65 const T* sorted_inputs_ptr = sorted_inputs.data() + b * num_inputs; in Compute()
112 const auto sorted_inputs = sorted_inputs_t.template flat<T>(); in Compute() local
116 ctx, sorted_inputs, values, sorted_inputs_t.dim_size(0), in Compute()
153 const auto sorted_inputs = sorted_inputs_t.template flat<T>(); in Compute() local
157 ctx, sorted_inputs, values, sorted_inputs_t.dim_size(0), in Compute()
|
| D | searchsorted_op.h | 33 const typename TTypes<T, 1>::ConstTensor& sorted_inputs,
44 const typename TTypes<T, 1>::ConstTensor& sorted_inputs,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_LowerBound.pbtxt | 5 name: "sorted_inputs"
27 Each set of rows with the same index in (sorted_inputs, values) is treated
29 `np.searchsorted(sorted_inputs, values, side='left')`.
|
| D | api_def_UpperBound.pbtxt | 5 name: "sorted_inputs"
27 Each set of rows with the same index in (sorted_inputs, values) is treated
29 `np.searchsorted(sorted_inputs, values, side='right')`.
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | lower_upper_bound_ops.cc | 37 const xla::XlaOp sorted_inputs = ctx->Input("sorted_inputs"); in BuildLowerUpperBoundOp() local
58 xla::Reshape(sorted_inputs, new_sorted_inputs_shape.dim_sizes()); in BuildLowerUpperBoundOp()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | UpperBound.pbtxt | 4 name: "sorted_inputs"
|
| D | LowerBound.pbtxt | 4 name: "sorted_inputs"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | LowerBound.pbtxt | 4 name: "sorted_inputs"
|
| D | UpperBound.pbtxt | 4 name: "sorted_inputs"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_cluster_formation.cc | 308 llvm::SmallVectorImpl<Operation*>* sorted_inputs) { in SortTPUReplicatedInputsByIndex() argument
326 sorted_inputs->assign(inputs.begin(), inputs.end()); in SortTPUReplicatedInputsByIndex()
328 sorted_inputs->begin(), sorted_inputs->end(), in SortTPUReplicatedInputsByIndex()
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | array_test.py | 179 sorted_inputs = math_ops.cumsum(
184 inputs_i = array_ops.gather(sorted_inputs, i)
|
| D | pfor.py | 2125 sorted_inputs = _flatten_first_two_dims(pfor_input.stacked_input(0))
2128 output = op_func(sorted_inputs, values, out_type)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 7026 Each set of rows with the same index in (sorted_inputs, values) is treated
7028 `np.searchsorted(sorted_inputs, values, side='left')`.
7046 Arg<TF_Tensor, [{2-D Tensor where each row is ordered.}]>:$sorted_inputs,
17390 Each set of rows with the same index in (sorted_inputs, values) is treated
17392 `np.searchsorted(sorted_inputs, values, side='right')`.
17410 Arg<TF_Tensor, [{2-D Tensor where each row is ordered.}]>:$sorted_inputs,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 2309 …argspec: "args=[\'sorted_inputs\', \'values\', \'out_type\', \'name\'], varargs=None, keywords=Non…
5273 …argspec: "args=[\'sorted_inputs\', \'values\', \'out_type\', \'name\'], varargs=None, keywords=Non…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 2309 …argspec: "args=[\'sorted_inputs\', \'values\', \'out_type\', \'name\'], varargs=None, keywords=Non…
5273 …argspec: "args=[\'sorted_inputs\', \'values\', \'out_type\', \'name\'], varargs=None, keywords=Non…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 30343 func LowerBound(scope *Scope, sorted_inputs tf.Output, values tf.Output, optional ...LowerBoundAttr…
30354 sorted_inputs, values,
44773 func UpperBound(scope *Scope, sorted_inputs tf.Output, values tf.Output, optional ...UpperBoundAttr…
44784 sorted_inputs, values,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 22855 name: "sorted_inputs"
58004 name: "sorted_inputs"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 33239 name: "sorted_inputs"
86465 name: "sorted_inputs"
|