Searched refs:kXlaInferredShapesAttrName (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass_test.cc | 210 const0_node->AddAttr(kXlaInferredShapesAttrName, in TEST()
284 kXlaInferredShapesAttrName, std::vector<PartialTensorShape>{shape}); in TEST_F()
443 kXlaInferredShapesAttrName, std::vector<PartialTensorShape>{shape}); in TEST_F()
459 kXlaInferredShapesAttrName, std::vector<PartialTensorShape>{shape}); in TEST_F()
594 kXlaInferredShapesAttrName, std::vector<PartialTensorShape>{shape}); in TEST_F()
610 kXlaInferredShapesAttrName, std::vector<PartialTensorShape>{shape}); in TEST_F()
721 kXlaInferredShapesAttrName, std::vector<PartialTensorShape>{shape}); in TEST_F()
869 kXlaInferredShapesAttrName, std::vector<PartialTensorShape>{shape}); in TEST_F()
951 kXlaInferredShapesAttrName, std::vector<PartialTensorShape>{shape}); in TEST_F()
|
| D | encapsulate_util_test.cc | 46 TF_CHECK_OK(GetNodeAttr(add_node->attrs(), kXlaInferredShapesAttrName, in TEST()
|
| D | encapsulate_util.h | 30 extern const char kXlaInferredShapesAttrName[];
|
| D | encapsulate_util.cc | 307 const char kXlaInferredShapesAttrName[] = "_xla_inferred_shapes"; variable
340 n->AddAttr(kXlaInferredShapesAttrName, output_shapes); in PerformStaticShapeInferenceBeforeEncapsulation()
|
| D | extract_outside_compilation_pass.cc | 289 if (!GetNodeAttr(e->src()->attrs(), kXlaInferredShapesAttrName, &shapes) in GetInferredInputShapes()
2251 n->ClearAttr(kXlaInferredShapesAttrName); in operator ()()
|