Searched refs:Prod (Results 1 - 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/contrib/reduce_slice_ops/kernels/
H A Dreduce_slice_ops.h25 #define Prod(a, b) ((a) * (b)) macro
63 func(Prod, functor::reduce_functions::one, ##__VA_ARGS__) func( \
/external/tensorflow/tensorflow/cc/gradients/
H A Ddata_flow_grad.cc82 scope, Range(scope, zero, Prod(scope, partitions_shape, zero), one),
H A Dmath_grad.cc724 auto group_size = SafeDivHelper(scope, Prod(scope, input_shape, zero),
725 Prod(scope, output_shape, zero));
845 // and we do a Prod operation on the axis 1, we will obtain [[105, 192]].
914 auto reduced_num = Prod(cpu_scope, Gather(scope, input_shape, reduced), 0);
917 auto other_num = Prod(cpu_scope, Gather(scope, input_shape, other), 0);
1007 REGISTER_GRADIENT_OP("Prod", ProdGrad);
H A Dmath_grad_test.cc44 using ops::Prod;
898 TEST_F(NaryGradTest, Prod) {
901 auto y = Prod(scope_, x, {1});
/external/swiftshader/third_party/LLVM/lib/Analysis/
H A DScalarEvolutionExpander.cpp782 Value *Prod = 0; local
786 if (!Prod) {
788 Prod = expand(Op);
792 Prod = InsertNoopCastOfTo(Prod, Ty);
793 Prod = InsertBinop(Instruction::Sub, Constant::getNullValue(Ty), Prod);
798 Prod = InsertNoopCastOfTo(Prod, Ty);
800 if (isa<Constant>(Prod)) st
[all...]
/external/llvm/lib/Analysis/
H A DScalarEvolutionExpander.cpp751 Value *Prod = nullptr; local
754 if (!Prod) {
756 Prod = expand(Op);
759 Prod = InsertNoopCastOfTo(Prod, Ty);
760 Prod = InsertBinop(Instruction::Sub, Constant::getNullValue(Ty), Prod);
764 Prod = InsertNoopCastOfTo(Prod, Ty);
766 if (isa<Constant>(Prod)) st
[all...]
/external/tensorflow/tensorflow/core/kernels/
H A Dreduction_gpu_kernels.cu.h70 struct Prod { struct in namespace:tensorflow::functor
79 struct Prod<std::complex<float>> { struct in namespace:tensorflow::functor
89 struct Prod<std::complex<double>> { struct in namespace:tensorflow::functor
731 (std::is_same<Op, Prod<T>>::value ||
968 ReduceImpl<T, Prod<T>, T*, T*, ReductionAxes>(
972 Prod<T>());
/external/swiftshader/third_party/LLVM/lib/Transforms/InstCombine/
H A DInstCombineCompares.cpp771 // Compute Prod = CI * DivRHS. We are essentially solving an equation
775 Constant *Prod = ConstantExpr::getMul(CmpRHS, DivRHS); local
780 bool ProdOV = (DivIsSigned ? ConstantExpr::getSDiv(Prod, DivRHS) :
781 ConstantExpr::getUDiv(Prod, DivRHS)) != CmpRHS;
788 ConstantInt *RangeSize = DivI->isExact() ? getOne(Prod) : DivRHS;
802 LoBound = Prod;
816 LoBound = Prod; // e.g. X/5 op 3 --> [15, 20)
819 HiOverflow = AddWithOverflow(HiBound, Prod, RangeSize, true);
822 HiBound = AddOne(Prod);
842 HiBound = AddOne(Prod);
[all...]
/external/llvm/lib/Transforms/InstCombine/
H A DInstCombineCompares.cpp1212 // Compute Prod = CI * DivRHS. We are essentially solving an equation
1216 Constant *Prod = ConstantExpr::getMul(CmpRHS, DivRHS); local
1221 bool ProdOV = (DivIsSigned ? ConstantExpr::getSDiv(Prod, DivRHS) :
1222 ConstantExpr::getUDiv(Prod, DivRHS)) != CmpRHS;
1229 ConstantInt *RangeSize = DivI->isExact() ? getOne(Prod) : DivRHS;
1243 LoBound = Prod;
1256 LoBound = Prod; // e.g. X/5 op 3 --> [15, 20)
1259 HiOverflow = AddWithOverflow(HiBound, Prod, RangeSize, true);
1262 HiBound = AddOne(Prod);
1282 HiBound = AddOne(Prod);
[all...]
/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dconstant_folding_test.cc1188 Output p = ops::Prod(scope.WithOpName("p"), v, i);
/external/tensorflow/tensorflow/compiler/tests/
H A Drandomized_tests.cc2445 TEST_F(OpTest, Prod) {
2451 return ExpectTfAndXlaOutputsAreClose(OpTestBuilder("Prod")

Completed in 369 milliseconds