Searched refs:output_attr (Results 1 – 7 of 7) sorted by relevance
642 output_attr = attr_value_pb2.AttrValue()643 output_attr.list.i.extend(hostmem)644 ret[0].op._set_attr("_output_hostmem", output_attr) # pylint: disable=protected-access772 output_attr = attr_value_pb2.AttrValue()773 output_attr.list.i.extend(hostmem)774 ret[0].op._set_attr("_output_hostmem", output_attr) # pylint: disable=protected-access
627 const auto& output_attr = output_node_def.attr(); in TEST_F() local628 EXPECT_EQ(2.0, output_attr.at("float_attr").f()); in TEST_F()629 EXPECT_EQ("Hello World", output_attr.at("str_attr").s()); in TEST_F()630 EXPECT_EQ(17, output_attr.at("int_attr").i()); in TEST_F()631 EXPECT_EQ(true, output_attr.at("bool_attr").b()); in TEST_F()633 const auto& list = output_attr.at("list_string_attr").list(); in TEST_F()642 const auto& list = output_attr.at("list_float_attr").list(); in TEST_F()649 const auto& list = output_attr.at("list_int_attr").list(); in TEST_F()
110 TfLiteTensor& output_attr = in Prepare() local112 tflite::SetTensorToDynamic(&output_attr); in Prepare()
200 TfLiteTensor& output_attr = in Prepare() local202 tflite::SetTensorToDynamic(&output_attr); in Prepare()
146 absl::InlinedVector<AllocatorAttributes, 4> output_attr(n->num_outputs()); in Compile() local147 params.output_attr_array = output_attr.data(); in Compile()
452 const auto output_attr = params_->output_attr_array == nullptr in forward_input_to_output_with_shape() local457 output_shape, output_memory_type(output_index), output_attr); in forward_input_to_output_with_shape()499 const AllocatorAttributes& output_attr) { in forward_input() argument554 if (!output_attr.IsEqualOrLessRestrictiveThan(input_attr)) { in forward_input()
853 const AllocatorAttributes& output_attr) TF_MUST_USE_RESULT;