HomeSort by relevance Sort by last modified time
    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