| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| runtime_fft.cc | 29 const void* run_options_ptr, void* out, void* operand, int32 fft_type,
35 operand, fft_type, fft_rank, input_batch,
|
| runtime_fft.h | 25 void* operand, tensorflow::int32 fft_type, tensorflow::int32 fft_rank,
|
| runtime_fft_impl.h | 180 int32 fft_type, int64 input_batch, int64 fft_length0,
182 CHECK(::xla::FftType_IsValid(fft_type)) << fft_type;
183 switch (fft_type) {
207 LOG(FATAL) << "Unsupported FFT type: " << fft_type;
215 int32 fft_type, int32 fft_rank, int64 input_batch,
220 device, out, operand, fft_type, input_batch, fft_length0, 0, 0);
223 internal::EigenFftWithRank<2, EigenDevice>(device, out, operand, fft_type,
228 internal::EigenFftWithRank<3, EigenDevice>(device, out, operand, fft_type,
|
| ir_emitter.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| fft_ops.cc | 41 explicit GenericFftOp(OpKernelConstruction* ctx, FftType fft_type,
43 : XlaOpKernel(ctx), fft_type_(fft_type), fft_rank_(fft_rank) {}
83 : GenericFftOp(ctx, /*fft_type=*/FftType::FFT, /*fft_rank=*/FFTRank) {}
93 : GenericFftOp(ctx, /*fft_type=*/FftType::IFFT, /*fft_rank=*/FFTRank) {}
103 : GenericFftOp(ctx, /*fft_type=*/FftType::RFFT, /*fft_rank=*/FFTRank) {}
113 : GenericFftOp(ctx, /*fft_type=*/FftType::IRFFT, /*fft_rank=*/FFTRank) {}
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| fft_thunk.h | 66 FftThunk(FftType fft_type, tensorflow::gtl::ArraySlice<int64> fft_length,
|
| fft_thunk.cc | 111 FftThunk::FftThunk(FftType fft_type,
118 fft_type_(FftTypeToSeType(fft_type)),
|
| ir_emitter_unnested.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo.proto | 128 xla.FftType fft_type = 31;
|
| shape_inference.h | 119 const Shape& in, FftType fft_type,
|
| hlo_instruction.h | 262 // Creates an FFT op, of the type indicated by fft_type.
264 const Shape& shape, HloInstruction* operand, FftType fft_type,
1092 FftType fft_type() const { function in class:xla::HloInstruction
[all...] |
| shape_inference.cc | [all...] |
| hlo_instruction.cc | 148 instruction->fft_type_ = proto.fft_type();
342 const Shape& shape, HloInstruction* operand, FftType fft_type,
346 instruction->fft_type_ = fft_type;
[all...] |
| hlo_verifier.cc | 87 ShapeInference::InferFftShape(fft->operand(0)->shape(), fft->fft_type(),
[all...] |
| user_computation.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
| hlo_parser_test.cc | 591 ROOT %fft = c64[8,32]{1,0} fft(c64[8,32]{1,0} %input), fft_type=FFT, fft_length={32}
603 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}
615 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}
627 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_length={64,128,64}
[all...] |
| hlo_parser.cc | 702 optional<FftType> fft_type; local
704 attrs["fft_type"] = {/*required=*/true, AttrTy::kFftType, &fft_type};
712 shape, operands[0], *fft_type, *fft_length));
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/ |
| xla_data.proto | 514 FftType fft_type = 1;
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| computation_builder.cc | 734 const ComputationDataHandle& operand, const FftType fft_type,
739 request->set_fft_type(fft_type);
[all...] |
| computation_builder.h | 419 FftType fft_type,
[all...] |
