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