| /external/tensorflow/tensorflow/core/kernels/ |
| D | argmax_op.cc | 45 class ArgOp : public OpKernel { class
47 explicit ArgOp(OpKernelConstruction* context) : OpKernel(context) {} in ArgOp() function in tensorflow::ArgOp
111 TF_DISALLOW_COPY_AND_ASSIGN(ArgOp);
116 : public ArgOp<Device, T, Tout, functor::ArgMax<Device, T, Tout> > {
119 : ArgOp<Device, T, Tout, functor::ArgMax<Device, T, Tout> >(context) {} in ArgMaxOp()
124 : public ArgOp<Device, T, Tout, functor::ArgMin<Device, T, Tout> > {
127 : ArgOp<Device, T, Tout, functor::ArgMin<Device, T, Tout> >(context) {} in ArgMinOp()
|
| D | function_ops.cc | 39 ArgOp::ArgOp(OpKernelConstruction* ctx) : OpKernel(ctx) { in ArgOp() function in tensorflow::ArgOp
44 void ArgOp::Compute(OpKernelContext* ctx) { in Compute()
87 REGISTER_SYSTEM_KERNEL_BUILDER(Name(kArgOp).Device(DEVICE_CPU), ArgOp);
88 REGISTER_SYSTEM_KERNEL_BUILDER(Name(kDeviceArgOp).Device(DEVICE_CPU), ArgOp);
100 Name(kArgOp).Device(DEVICE_GPU).TypeConstraint<type>("T"), ArgOp);
107 ArgOp);
109 Name(kDeviceArgOp).Device(DEVICE_GPU).TypeConstraint<int32>("T"), ArgOp);
116 ArgOp);
122 ArgOp);
125 Name(kArgOp).Device(DEVICE_GPU).TypeConstraint<Variant>("T"), ArgOp);
|
| D | function_ops.h | 29 class ArgOp : public OpKernel {
31 explicit ArgOp(OpKernelConstruction* ctx);
41 TF_DISALLOW_COPY_AND_ASSIGN(ArgOp);
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_ops.h | 156 Name(kArgOp).Device(DEVICE).TypeConstraint("T", TYPES), ArgOp); \
161 ArgOp); \
163 Name(kArgOp).Device(DEVICE).TypeConstraint<Variant>("T"), ArgOp); \
226 ArgOp); \
|
| D | encapsulate_subgraphs_pass_test.cc | 467 Node* ArgOp(int index, DataType type, const GraphDefBuilder::Options& opts) { in ArgOp() function
|
| /external/swiftshader/third_party/llvm-10.0/llvm/lib/Transforms/IPO/ |
| D | Attributor.cpp | 2482 const Value *ArgOp = ICS.getArgOperand(i); in updateImpl() local
2483 if (!ArgOp->getType()->isPointerTy()) in updateImpl()
2488 bool IsAliasing = !AAR->isNoAlias(&getAssociatedValue(), ArgOp); in updateImpl()
2492 << AAR->isNoAlias(&getAssociatedValue(), ArgOp) << " " in updateImpl()
2493 << getAssociatedValue() << " " << *ArgOp << " => " in updateImpl()
4255 Value *ArgOp = ACS.getCallArgOperand(getArgNo()); in updateImpl() local
4256 if (!ArgOp) in updateImpl()
4264 if (auto *C = dyn_cast<Constant>(ArgOp)) in updateImpl()
4267 return checkAndUpdate(A, *this, *ArgOp, SimplifiedAssociatedValue); in updateImpl()
|
| /external/llvm-project/llvm/lib/Transforms/IPO/ |
| D | OpenMPOpt.cpp | 1294 Value *ArgOp = CI->getArgOperand(ArgNo); in collectGlobalThreadIdArguments() local
1295 if (CI == &RefCI || GTIdArgs.count(ArgOp) || in collectGlobalThreadIdArguments()
1297 *ArgOp, &OMPInfoCache.RFIs[OMPRTL___kmpc_global_thread_num])) in collectGlobalThreadIdArguments()
|
| D | AttributorAttributes.cpp | 2523 const Value *ArgOp = CB.getArgOperand(OtherArgNo); in mayAliasWithArgument() local
2524 if (!ArgOp->getType()->isPtrOrPtrVectorTy()) in mayAliasWithArgument()
2551 bool IsAliasing = !AAR || !AAR->isNoAlias(&getAssociatedValue(), ArgOp); in mayAliasWithArgument()
2554 << getAssociatedValue() << " " << *ArgOp << " => " in mayAliasWithArgument()
4678 Value &ArgOp = ACSArgPos.getAssociatedValue(); in updateImpl() local
4680 if (auto *C = dyn_cast<Constant>(&ArgOp)) in updateImpl()
4683 return checkAndUpdate(A, *this, ArgOp, SimplifiedAssociatedValue); in updateImpl()
6703 const Value *ArgOp = CB.getArgOperand(ArgNo); in categorizeArgumentPointerLocations() local
6704 if (!ArgOp->getType()->isPtrOrPtrVectorTy()) in categorizeArgumentPointerLocations()
6717 categorizePtrValue(A, CB, *ArgOp, AccessedLocs, Changed); in categorizeArgumentPointerLocations()
|
| /external/llvm/lib/Transforms/Vectorize/ |
| D | LoopVectorize.cpp | 3281 for (auto &ArgOp : CI->arg_operands()) in getVectorCallCost() local
3282 ScalarTys.push_back(ArgOp->getType()); in getVectorCallCost()
|
| /external/swiftshader/third_party/llvm-10.0/llvm/lib/Transforms/Vectorize/ |
| D | LoopVectorize.cpp | 3264 for (auto &ArgOp : CI->arg_operands()) in getVectorCallCost() local
3265 ScalarTys.push_back(ArgOp->getType()); in getVectorCallCost()
|
| /external/llvm-project/llvm/lib/Transforms/Vectorize/ |
| D | LoopVectorize.cpp | 3683 for (auto &ArgOp : CI->arg_operands()) in getVectorCallCost() local
3684 ScalarTys.push_back(ArgOp->getType()); in getVectorCallCost()
|