OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:ArgOp
(Results
1 - 3
of
3
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
argmax_op.cc
44
class
ArgOp
: public OpKernel {
46
explicit
ArgOp
(OpKernelConstruction* context) : OpKernel(context) {}
96
"
ArgOp
: Unhandled input dimensions: ", input_dims));
102
TF_DISALLOW_COPY_AND_ASSIGN(
ArgOp
);
107
: public
ArgOp
<Device, T, Tout, functor::ArgMax<Device, T, Tout> > {
110
:
ArgOp
<Device, T, Tout, functor::ArgMax<Device, T, Tout> >(context) {}
115
: public
ArgOp
<Device, T, Tout, functor::ArgMin<Device, T, Tout> > {
118
:
ArgOp
<Device, T, Tout, functor::ArgMin<Device, T, Tout> >(context) {}
function_ops.cc
37
class
ArgOp
: public OpKernel {
39
explicit
ArgOp
(OpKernelConstruction* ctx) : OpKernel(ctx) {
62
TF_DISALLOW_COPY_AND_ASSIGN(
ArgOp
);
92
REGISTER_SYSTEM_KERNEL_BUILDER(Name("_Arg").Device(DEVICE_CPU),
ArgOp
);
98
Name("_Arg").Device(DEVICE_SYCL).TypeConstraint<type>("T"),
ArgOp
);
104
ArgOp
);
121
Name("_Arg").Device(DEVICE_GPU).TypeConstraint<type>("T"),
ArgOp
);
127
ArgOp
);
134
ArgOp
);
/external/tensorflow/tensorflow/compiler/jit/
encapsulate_subgraphs_pass_test.cc
393
Node*
ArgOp
(int index, DataType type, const GraphDefBuilder::Options& opts) {
[
all
...]
Completed in 149 milliseconds