Home
last modified time | relevance | path

Searched refs:ArgOp (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dfunction_ops.cc38 ArgOp::ArgOp(OpKernelConstruction* ctx) : OpKernel(ctx) { in ArgOp() function in tensorflow::ArgOp
43 void ArgOp::Compute(OpKernelContext* ctx) { in Compute()
71 REGISTER_SYSTEM_KERNEL_BUILDER(Name(kArgOp).Device(DEVICE_CPU), ArgOp);
72 REGISTER_SYSTEM_KERNEL_BUILDER(Name(kDeviceArgOp).Device(DEVICE_CPU), ArgOp);
79 Name(kArgOp).Device(DEVICE_SYCL).TypeConstraint<type>("T"), ArgOp);
85 ArgOp);
101 Name(kArgOp).Device(DEVICE_GPU).TypeConstraint<type>("T"), ArgOp);
108 ArgOp);
110 Name(kDeviceArgOp).Device(DEVICE_GPU).TypeConstraint<int32>("T"), ArgOp);
117 ArgOp);
[all …]
Dargmax_op.cc44 class ArgOp : public OpKernel { class
46 explicit ArgOp(OpKernelConstruction* context) : OpKernel(context) {} in ArgOp() function in tensorflow::ArgOp
106 TF_DISALLOW_COPY_AND_ASSIGN(ArgOp);
111 : public ArgOp<Device, T, Tout, functor::ArgMax<Device, T, Tout> > {
114 : ArgOp<Device, T, Tout, functor::ArgMax<Device, T, Tout> >(context) {} in ArgMaxOp()
119 : public ArgOp<Device, T, Tout, functor::ArgMin<Device, T, Tout> > {
122 : ArgOp<Device, T, Tout, functor::ArgMin<Device, T, Tout> >(context) {} in ArgMinOp()
Dfunction_ops.h29 class ArgOp : public OpKernel {
31 explicit ArgOp(OpKernelConstruction* ctx);
41 TF_DISALLOW_COPY_AND_ASSIGN(ArgOp);
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_ops.h211 Name(kArgOp).Device(DEVICE).TypeConstraint("T", TYPES), ArgOp); \
216 ArgOp); \
218 Name(kArgOp).Device(DEVICE).TypeConstraint<Variant>("T"), ArgOp); \
277 ArgOp); \
Dencapsulate_subgraphs_pass_test.cc476 Node* ArgOp(int index, DataType type, const GraphDefBuilder::Options& opts) { in ArgOp() function
/external/llvm/lib/Transforms/Vectorize/
DLoopVectorize.cpp3281 for (auto &ArgOp : CI->arg_operands()) in getVectorCallCost() local
3282 ScalarTys.push_back(ArgOp->getType()); in getVectorCallCost()
/external/swiftshader/third_party/llvm-7.0/llvm/lib/Transforms/Vectorize/
DLoopVectorize.cpp3137 for (auto &ArgOp : CI->arg_operands()) in getVectorCallCost() local
3138 ScalarTys.push_back(ArgOp->getType()); in getVectorCallCost()