| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_fft_impl.h | 173 int32 fft_type, int64 input_batch, int64 fft_length0, in EigenFftWithRank() argument
175 switch (fft_type) { in EigenFftWithRank()
208 int32 fft_type, int32 fft_rank, int64 input_batch, in EigenFftImpl() argument
213 device, out, operand, fft_type, input_batch, fft_length0, 0, 0); in EigenFftImpl()
216 internal::EigenFftWithRank<2, EigenDevice>(device, out, operand, fft_type, in EigenFftImpl()
221 internal::EigenFftWithRank<3, EigenDevice>(device, out, operand, fft_type, in EigenFftImpl()
|
| D | runtime_single_threaded_fft.cc | 26 const void* run_options_ptr, void* out, void* operand, int32 fft_type, in __xla_cpu_runtime_EigenSingleThreadedFft() argument
29 tensorflow::xla::EigenFftImpl(Eigen::DefaultDevice(), out, operand, fft_type, in __xla_cpu_runtime_EigenSingleThreadedFft()
|
| D | runtime_fft.cc | 29 const void* run_options_ptr, void* out, void* operand, int32 fft_type, in __xla_cpu_runtime_EigenFft() argument
35 operand, fft_type, fft_rank, input_batch, in __xla_cpu_runtime_EigenFft()
|
| D | runtime_fft.h | 25 void* operand, tensorflow::int32 fft_type, tensorflow::int32 fft_rank,
|
| D | runtime_single_threaded_fft.h | 25 void* operand, tensorflow::int32 fft_type, tensorflow::int32 fft_rank,
|
| D | ir_emitter.cc | 1279 llvm::FunctionType* fft_type = llvm::FunctionType::get( in HandleFft() local
1292 module_->getOrInsertFunction(fn_name, fft_type).getCallee()); in HandleFft()
1300 BitCast(operand_address, int8_ptr_type), b_.getInt32(fft->fft_type()), in HandleFft()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | fft_ops.cc | 42 explicit GenericFftOp(OpKernelConstruction* ctx, FftType fft_type, in GenericFftOp() argument
44 : XlaOpKernel(ctx), fft_type_(fft_type), fft_rank_(fft_rank) {} in GenericFftOp()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | fft_thunk.cc | 95 FftThunk::FftThunk(FftType fft_type, absl::Span<const int64> fft_length, in FftThunk() argument
101 fft_type_(FftTypeToSeType(fft_type)), in FftThunk()
|
| D | fft_thunk.h | 65 FftThunk(FftType fft_type, absl::Span<const int64> fft_length,
|
| D | ir_emitter_unnested.cc | 1877 inst->fft_type(), inst->fft_length(), in BuildFftThunk()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.h | 116 static StatusOr<Shape> InferFftShape(const Shape& in, FftType fft_type,
|
| D | hlo_parser_test.cc | 706 ROOT %fft = c64[8,32]{1,0} fft(c64[8,32]{1,0} %input), fft_type=FFT, fft_length={32} in CreateTestCases()
718 ROOT %fft = c64[5,8,32]{2,1,0} fft(c64[5,8,32]{2,1,0} %input), fft_type=IFFT, fft_length={8,32} in CreateTestCases()
730 ROOT %fft = c64[5,64,17]{2,1,0} fft(f32[5,64,32]{2,1,0} %input), fft_type=RFFT, fft_length={64,32} in CreateTestCases()
742 …ROOT %fft = f32[5,64,128,64]{3,2,1,0} fft(c64[5,64,128,33]{3,2,1,0} %input), fft_type=IRFFT, fft_l… in CreateTestCases()
|
| D | hlo_instructions.cc | 166 HloInstruction* operand, FftType fft_type, in HloFftInstruction() argument
168 : HloInstruction(HloOpcode::kFft, shape), fft_type_(fft_type) { in HloFftInstruction()
184 return {StrCat("fft_type=", FftType_Name(fft_type())), in ExtraAttributesToStringImpl()
193 return fft_type() == casted_other.fft_type() && in IdenticalSlowPath()
|
| D | hlo_instruction.h | 445 const Shape& shape, HloInstruction* operand, FftType fft_type,
1384 FftType fft_type() const;
|
| D | shape_inference.cc | 1819 const Shape& in, const FftType fft_type, in InferFftShape() argument
1832 switch (fft_type) { in InferFftShape()
1837 FftType_Name(fft_type), in InferFftShape()
1898 LOG(FATAL) << "Unexpected fft_type: " << fft_type; in InferFftShape()
|
| D | hlo_instruction.cc | 163 instruction = CreateFft(shape, operands(0), proto.fft_type(), in CreateFromProto()
791 const Shape& shape, HloInstruction* operand, FftType fft_type, in CreateFft() argument
793 return absl::make_unique<HloFftInstruction>(shape, operand, fft_type, in CreateFft()
3222 FftType HloInstruction::fft_type() const { in fft_type() function in xla::HloInstruction
3223 return Cast<HloFftInstruction>(this)->fft_type(); in fft_type()
|
| D | hlo_instructions.h | 105 FftType fft_type,
107 FftType fft_type() const { return fft_type_; } in fft_type() function
|
| D | hlo.proto | 144 xla.FftType fft_type = 31; field
|
| D | hlo_parser.cc | 1107 optional<FftType> fft_type; in ParseInstructionRhs() local
1109 attrs["fft_type"] = {/*required=*/true, AttrTy::kFftType, &fft_type}; in ParseInstructionRhs()
1117 shape, operands[0], *fft_type, *fft_length)); in ParseInstructionRhs()
|
| D | hlo_verifier.cc | 167 ShapeInference::InferFftShape(fft->operand(0)->shape(), fft->fft_type(), in HandleFft()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 419 XlaOp Fft(const XlaOp& operand, FftType fft_type,
812 friend XlaOp Fft(const XlaOp& operand, FftType fft_type,
1349 XlaOp Fft(const XlaOp& operand, FftType fft_type,
|
| D | xla_builder.cc | 1264 XlaOp XlaBuilder::Fft(const XlaOp& operand, const FftType fft_type, in Fft() argument
1270 operand_shape, fft_type, fft_length)); in Fft()
1272 instr.set_fft_type(fft_type); in Fft()
3024 XlaOp Fft(const XlaOp& operand, FftType fft_type, in Fft() argument
3026 return operand.builder()->Fft(operand, fft_type, fft_length); in Fft()
|