| /external/tensorflow/tensorflow/core/kernels/ |
| D | bias_op_gpu.h | 65 return (algorithm_ != BiasAddGradGPUMode::kInvalid && in is_valid()
68 BiasAddGradGPUMode algorithm() const { return algorithm_; } in algorithm()
69 void set_algorithm(BiasAddGradGPUMode val) { algorithm_ = val; } in set_algorithm()
74 BiasAddGradGPUMode algorithm_ = BiasAddGradGPUMode::kInvalid;
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | model_dataset_op.cc | 53 algorithm_(algorithm), in Dataset()
103 b->BuildAttrValue(static_cast<int64>(algorithm_), &algorithm_attr); in AsGraphDefInternal()
194 model_->OptimizeLoop(dataset()->algorithm_, cpu_budget_, in EnsureOptimizationLoopThreadStarted()
239 const model::AutotuneAlgorithm algorithm_; member in tensorflow::data::ModelDatasetOp::Dataset
250 algorithm_ = model::AutotuneAlgorithm(algorithm); in ModelDatasetOp()
252 algorithm_ = model::AutotuneAlgorithm::HILL_CLIMB; in ModelDatasetOp()
270 *output = new ModelDatasetOp::Dataset(ctx, input, algorithm_, cpu_budget_, in MakeDataset()
|
| D | model_dataset_op.h | 44 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::__anone4066a570111::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 | 777 return algorithm_.has_value() && in is_valid()
781 AlgorithmDesc algorithm() const { return *algorithm_; } in algorithm()
782 void set_algorithm(AlgorithmDesc val) { algorithm_ = val; } in set_algorithm()
791 absl::optional<AlgorithmDesc> algorithm_;
817 explicit AlgorithmConfig(AlgorithmDesc algorithm) : algorithm_(algorithm) {} in AlgorithmConfig()
819 : algorithm_(algorithm), scratch_size_(scratch_size) {} in AlgorithmConfig()
821 : algorithm_(algorithm), algorithm_no_scratch_(algorithm_no_scratch) {} in AlgorithmConfig()
822 absl::optional<AlgorithmDesc> algorithm() const { return algorithm_; } in algorithm()
823 void set_algorithm(AlgorithmDesc val) { algorithm_ = val; } in set_algorithm()
833 return this->algorithm_ == other.algorithm_ &&
[all …]
|
| D | blas.cc | 72 return absl::StrCat(algorithm_); in ToString()
|
| D | blas.h | 167 AlgorithmType algorithm() const { return algorithm_; } in algorithm()
168 void set_algorithm(AlgorithmType val) { algorithm_ = val; } in set_algorithm()
174 AlgorithmType algorithm_ = kDefaultAlgorithm; variable
180 AlgorithmConfig() : algorithm_(kDefaultAlgorithm) {} in AlgorithmConfig()
181 explicit AlgorithmConfig(AlgorithmType algorithm) : algorithm_(algorithm) {} in AlgorithmConfig()
182 AlgorithmType algorithm() const { return algorithm_; } in algorithm()
183 void set_algorithm(AlgorithmType val) { algorithm_ = val; } in set_algorithm()
185 return this->algorithm_ == other.algorithm_;
193 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 | 145 ModuleSchedulerAlgorithm algorithm_; variable
|
| D | hlo_memory_scheduler.cc | 731 : size_function_(size_function), algorithm_(algorithm) {} in HloMemoryScheduler()
735 ScheduleModule(module, size_function_, algorithm_)); in Run()
|
| D | heap_simulator.h | 213 const std::unique_ptr<HeapAlgorithm<HloValue>> algorithm_; variable
|
| D | hlo_instructions.h | 1934 RandomAlgorithm algorithm() const { return algorithm_; } in algorithm()
1948 RandomAlgorithm algorithm_; variable
|
| D | hlo_instructions.cc | 3192 algorithm_(algorithm) { in HloRngBitGeneratorInstruction()
3198 proto.set_rng_algorithm(algorithm_); in ToProto()
3204 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::__anon371466c50111::ChecksumFunction
|