Home
last modified time | relevance | path

Searched refs:output_attr (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dfunctional_ops.py642 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-access
772 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
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator_test.cc627 const auto& output_attr = output_node_def.attr(); in TEST_F() local
628 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()
/external/libtextclassifier/utils/tflite/
Dtoken_encoder.cc110 TfLiteTensor& output_attr = in Prepare() local
112 tflite::SetTensorToDynamic(&output_attr); in Prepare()
Dtext_encoder.cc200 TfLiteTensor& output_attr = in Prepare() local
202 tflite::SetTensorToDynamic(&output_attr); in Prepare()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dgraph_compiler.cc146 absl::InlinedVector<AllocatorAttributes, 4> output_attr(n->num_outputs()); in Compile() local
147 params.output_attr_array = output_attr.data(); in Compile()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.cc452 const auto output_attr = params_->output_attr_array == nullptr in forward_input_to_output_with_shape() local
457 output_shape, output_memory_type(output_index), output_attr); in forward_input_to_output_with_shape()
499 const AllocatorAttributes& output_attr) { in forward_input() argument
554 if (!output_attr.IsEqualOrLessRestrictiveThan(input_attr)) { in forward_input()
Dop_kernel.h853 const AllocatorAttributes& output_attr) TF_MUST_USE_RESULT;