| /external/tensorflow/tensorflow/core/kernels/ |
| D | bias_op_gpu.h | 66 return (algorithm_ != BiasAddGradGPUMode::kInvalid && in is_valid()
69 BiasAddGradGPUMode algorithm() const { return algorithm_; } in algorithm()
70 void set_algorithm(BiasAddGradGPUMode val) { algorithm_ = val; } in set_algorithm()
75 BiasAddGradGPUMode algorithm_ = BiasAddGradGPUMode::kInvalid;
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | model_dataset_op.cc | 61 algorithm_(algorithm), in Dataset()
111 b->BuildAttrValue(static_cast<int64>(algorithm_), &algorithm_attr); in AsGraphDefInternal()
194 model_->OptimizeLoop(dataset()->algorithm_, cpu_budget_, in EnsureOptimizationLoopThreadStarted()
216 const model::AutotuneAlgorithm algorithm_; member in tensorflow::data::ModelDatasetOp::Dataset
239 algorithm_ = model::AutotuneAlgorithm(algorithm); in ModelDatasetOp()
241 algorithm_ = model::AutotuneAlgorithm::HILL_CLIMB; in ModelDatasetOp()
259 *output = new ModelDatasetOp::Dataset(ctx, input, algorithm_, cpu_budget_, in MakeDataset()
|
| D | model_dataset_op.h | 53 model::AutotuneAlgorithm algorithm_; variable
|
| /external/rust/crates/grpcio-sys/grpc/src/core/ext/filters/http/message_compress/ |
| D | message_decompress_filter.cc | 117 grpc_message_compression_algorithm algorithm_ = GRPC_MESSAGE_COMPRESS_NONE; member in grpc_core::__anona1411c1c0111::CallData
158 calld->algorithm_ = DecodeMessageCompressionAlgorithm(grpc_encoding->md); in OnRecvInitialMetadataReady()
187 if (calld->algorithm_ != GRPC_MESSAGE_COMPRESS_NONE) { in OnRecvMessageReady()
260 if (grpc_msg_decompress(algorithm_, &recv_slices_, &decompressed_slices) == in FinishRecvMessage()
266 algorithm_) in FinishRecvMessage()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 831 return algorithm_.has_value() && in is_valid()
835 AlgorithmDesc algorithm() const { return *algorithm_; } in algorithm()
836 void set_algorithm(AlgorithmDesc val) { algorithm_ = val; } in set_algorithm()
845 absl::optional<AlgorithmDesc> algorithm_;
871 explicit AlgorithmConfig(AlgorithmDesc algorithm) : algorithm_(algorithm) {} in AlgorithmConfig()
873 : algorithm_(algorithm), scratch_size_(scratch_size) {} in AlgorithmConfig()
875 : algorithm_(algorithm), algorithm_no_scratch_(algorithm_no_scratch) {} in AlgorithmConfig()
878 : algorithm_(algorithm), in AlgorithmConfig()
886 algorithm_ = AlgorithmDesc( in AlgorithmConfig()
904 absl::optional<AlgorithmDesc> algorithm() const { return algorithm_; } in algorithm()
[all …]
|
| D | blas.cc | 72 return absl::StrCat(algorithm_); in ToString()
|
| D | blas.h | 198 AlgorithmType algorithm() const { return algorithm_; }
199 void set_algorithm(AlgorithmType val) { algorithm_ = val; }
205 AlgorithmType algorithm_ = kDefaultAlgorithm;
211 AlgorithmConfig() : algorithm_(kDefaultAlgorithm) {}
212 explicit AlgorithmConfig(AlgorithmType algorithm) : algorithm_(algorithm) {}
213 AlgorithmType algorithm() const { return algorithm_; }
214 void set_algorithm(AlgorithmType val) { algorithm_ = val; }
216 return this->algorithm_ == other.algorithm_;
224 AlgorithmType algorithm_;
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | heap_simulator.cc | 342 algorithm_(std::move(algorithm)), in HeapSimulator()
375 algorithm_->Alloc(buffer, size); in Alloc()
388 algorithm_->Free(buffer, size); in Free()
400 algorithm_->ShareWith(buffer, shared, size_fn_(*shared)); in ShareBuffer()
407 Result<HloValue> result = algorithm_->Finish(); in Finish()
|
| D | hlo_memory_scheduler.h | 153 ModuleSchedulerAlgorithm algorithm_; variable
|
| D | hlo_memory_scheduler.cc | 747 : size_function_(size_function), algorithm_(algorithm) {} in HloMemoryScheduler()
751 ScheduleModule(module, size_function_, algorithm_)); in Run()
|
| D | heap_simulator.h | 213 const std::unique_ptr<HeapAlgorithm<HloValue>> algorithm_; variable
|
| D | hlo_instructions.h | 2037 RandomAlgorithm algorithm() const { return algorithm_; } in algorithm()
2051 RandomAlgorithm algorithm_; variable
|
| D | hlo_instructions.cc | 3357 algorithm_(algorithm) { in HloRngBitGeneratorInstruction()
3363 proto.set_rng_algorithm(algorithm_); in ToProto()
3369 return {StrCat("algorithm=", RandomAlgorithmToString(algorithm_))}; in ExtraAttributesToStringImpl()
|
| /external/pigweed/pw_kvs/ |
| D | key_value_store_binary_format_test.cc | 48 algorithm_(algorithm) {} in ChecksumFunction()
53 state_ = algorithm_(data, state_); in Update()
58 State (&algorithm_)(std::span<const byte>, State); member in pw::kvs::__anond47ed32a0111::ChecksumFunction
|