Home
last modified time | relevance | path

Searched refs:ArgMin (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dargmax_op.cc119 : public ArgOp<Device, T, Tout, functor::ArgMin<Device, T, Tout> > {
122 : ArgOp<Device, T, Tout, functor::ArgMin<Device, T, Tout> >(context) {} in ArgMinOp()
164 void ArgMin<GPUDevice, T, Tout>::Reduce##Dims( \
182 extern template struct ArgMin<GPUDevice, T, int64>; \
184 extern template struct ArgMin<GPUDevice, T, int32>;
Dargmax_op_gpu.cu.cc29 template struct functor::ArgMin<GPUDevice, T, int64>; \
31 template struct functor::ArgMin<GPUDevice, T, int32>;
Dargmax_op.h47 struct ArgMin { struct
/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_ArgMin.pbtxt2 graph_op_name: "ArgMin"
4 name: "math.ArgMin"
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Darithmetic_test.cc33 ArgMin(x, S32, /*axis=*/0); in XLA_TEST_F()
42 ArgMin(x, S32, /*axis=*/1); in XLA_TEST_F()
Darithmetic.h66 XlaOp ArgMin(XlaOp input, PrimitiveType output_type, int axis);
Darithmetic.cc182 XlaOp ArgMin(XlaOp input, PrimitiveType output_type, int axis) { in ArgMin() function
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_ArgMin.pbtxt2 graph_op_name: "ArgMin"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_ArgMin.pbtxt2 graph_op_name: "ArgMin"
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dindex_ops.cc68 output = xla::ArgMin(input, index_xla_type, axis); in Compile()
/external/swiftshader/third_party/llvm-7.0/llvm/lib/Support/Unix/
DProgram.inc462 static long ArgMin = _POSIX_ARG_MAX;
469 else if (EffectiveArgMax < ArgMin)
470 EffectiveArgMax = ArgMin;
/external/walt/ios/WALT/
DDragLatencyController.mm117 static size_t ArgMin(const std::vector<T>& x) { function
155 return ArgMin(deviations) * kShiftStep;
/external/tensorflow/tensorflow/compiler/tf2xla/g3doc/
Dcpu_supported_ops.md18 `ArgMin` | `Tidx={int32,int64}`<br>`output_type={int32,int64}`<br>`T={…
Dgpu_supported_ops.md18 `ArgMin` | `Tidx={int32,int64}`<br>`output_type={int32,int64}`<br>`T={…
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator.cc1559 class ArgMin : public BuiltinOperator<ArgMinOperator, ::tflite::ArgMinOptions, class
2433 ops.push_back(MakeUnique<ArgMin>(::tflite::BuiltinOperator_ARG_MIN, in BuildOperatorList()
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc399 HANDLE_OPERATORTYPENAME_CASE(ArgMin) in OperatorTypeName()
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc1059 TEST_F(OpTest, ArgMin) { in TEST_F() argument
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v0.pbtxt2932 name: "ArgMin"
2966 name: "ArgMin"
3001 name: "ArgMin"
3037 name: "ArgMin"
3074 name: "ArgMin"
Dops_history.v2.pbtxt6232 name: "ArgMin"
6282 name: "ArgMin"
6345 name: "ArgMin"
6410 name: "ArgMin"
6476 name: "ArgMin"
Dops_history.v1.pbtxt6232 name: "ArgMin"
6282 name: "ArgMin"
6345 name: "ArgMin"
6410 name: "ArgMin"
6476 name: "ArgMin"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt168 name: "ArgMin"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt168 name: "ArgMin"
/external/tensorflow/tensorflow/go/op/
Dwrappers.go32363 func ArgMin(scope *Scope, input tf.Output, dimension tf.Output, optional ...ArgMinAttr) (output tf.… func
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt1992 name: "ArgMin"