Home
last modified time | relevance | path

Searched refs:TypeAttrValue (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_jit_compiled_cpu_function_test.cc34 AttrValue TypeAttrValue(DataType type) { in TypeAttrValue() function
45 (*x->mutable_attr())["dtype"] = TypeAttrValue(DT_INT32); in SumGraph()
49 (*y->mutable_attr())["dtype"] = TypeAttrValue(DT_INT32); in SumGraph()
55 (*sum->mutable_attr())["T"] = TypeAttrValue(DT_INT32); in SumGraph()
138 TypeAttrValue(DT_INT32); in TEST()
Dtf2xla_test.cc36 AttrValue TypeAttrValue(DataType type) { in TypeAttrValue() function
47 (*x->mutable_attr())["dtype"] = TypeAttrValue(DT_INT32); in SumGraph()
51 (*y->mutable_attr())["dtype"] = TypeAttrValue(DT_INT32); in SumGraph()
57 (*sum->mutable_attr())["T"] = TypeAttrValue(DT_INT32); in SumGraph()