OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:ArgMin
(Results
1 - 9
of
9
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
argmax_op_gpu.cu.cc
29
template struct functor::
ArgMin
<GPUDevice, T, int64>; \
31
template struct functor::
ArgMin
<GPUDevice, T, int32>;
argmax_op.cc
115
: public ArgOp<Device, T, Tout, functor::
ArgMin
<Device, T, Tout> > {
118
: ArgOp<Device, T, Tout, functor::
ArgMin
<Device, T, Tout> >(context) {}
128
REGISTER_KERNEL_BUILDER(Name("
ArgMin
") \
140
REGISTER_KERNEL_BUILDER(Name("
ArgMin
") \
160
void
ArgMin
<GPUDevice, T, Tout>::Reduce##Dims( \
178
extern template struct
ArgMin
<GPUDevice, T, int64>; \
180
extern template struct
ArgMin
<GPUDevice, T, int32>;
199
REGISTER_KERNEL_BUILDER(Name("
ArgMin
") \
213
REGISTER_KERNEL_BUILDER(Name("
ArgMin
") \
argmax_op.h
47
struct
ArgMin
{
52
output.device(d) = input.
argmin
(dimension).template cast<Tout>(); \
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
index_ops.cc
70
XlaHelpers::
ArgMin
(b, ctx, input, input_shape, input_type(0),
96
REGISTER_XLA_OP(Name("
ArgMin
").CompileTimeConstInput("dimension"), XlaArgMinOp);
/external/tensorflow/tensorflow/compiler/tf2xla/
xla_helpers.h
85
// Sets `
argmin
` to the
argmin
of `input` along `axis`. `input_shape` and
87
// `output_type` is the dtype to use for `
argmin
`.
88
static Status
ArgMin
(xla::ComputationBuilder* builder,
93
xla::ComputationDataHandle*
argmin
);
xla_helpers.cc
192
Status XlaHelpers::
ArgMin
(xla::ComputationBuilder* builder,
197
xla::ComputationDataHandle*
argmin
) {
199
axis, /*is_min=*/true,
argmin
);
/external/walt/ios/WALT/
DragLatencyController.mm
117
static size_t
ArgMin
(const std::vector<T>& x) {
155
return
ArgMin
(deviations) * kShiftStep;
/external/tensorflow/tensorflow/compiler/tests/
randomized_tests.cc
[
all
...]
/external/tensorflow/tensorflow/go/op/
wrappers.go
[
all
...]
Completed in 287 milliseconds