Searched refs:SetFrontendAttributes (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | host_compute_ops.cc | 161 b->SetFrontendAttributes(attrs); in Compile()
221 b->SetFrontendAttributes(attrs); in Compile()
426 b->SetFrontendAttributes(attrs); in Compile()
478 b->SetFrontendAttributes(attrs); in Compile()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_communication.cc | 224 void SetFrontendAttributes(Operation* op, int32_t index, StringRef key, in SetFrontendAttributes() function
265 SetFrontendAttributes(send, index, key, operand.getType(), in CreateSendOp()
289 SetFrontendAttributes(recv, index, key, result_type, in CreateRecvOp()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 1197 b.SetFrontendAttributes(attributes); in TEST_F()
1219 b.SetFrontendAttributes(attributes); in TEST_F()
1227 b.SetFrontendAttributes(attributes); in TEST_F()
1236 b.SetFrontendAttributes(attributes); in TEST_F()
1259 b.SetFrontendAttributes(attributes); in TEST_F()
|
| D | xla_builder.h | 203 void SetFrontendAttributes(const FrontendAttributes& frontend_attributes) { in SetFrontendAttributes() function
1519 builder_->SetFrontendAttributes(saved_); in ~XlaScopedFrontendAttributesAssignment()
|