Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/
Dtf_framework_c_interface.cc56 AllocatorAttributes output_attr; in _mlir_ciface_tf_alloc() local
60 ctx->output_memory_type(output_index), output_attr); in _mlir_ciface_tf_alloc()
/external/tensorflow/tensorflow/compiler/mlir/lite/python/
Dsaved_model_to_tfl_flatbuffer.cc98 auto output_attr = tf_attrs.get("outputs"); in HandleInputOutputArraysWithModule() local
99 if (!output_attr) { in HandleInputOutputArraysWithModule()
102 auto output_names = output_attr.cast<mlir::StringAttr>().getValue(); in HandleInputOutputArraysWithModule()
/external/tensorflow/tensorflow/python/ops/
Dfunctional_ops.py987 output_attr = attr_value_pb2.AttrValue()
988 output_attr.list.i.extend(hostmem)
989 ret[0].op._set_attr("_output_hostmem", output_attr) # pylint: disable=protected-access
1117 output_attr = attr_value_pb2.AttrValue()
1118 output_attr.list.i.extend(hostmem)
1119 ret[0].op._set_attr("_output_hostmem", output_attr) # pylint: disable=protected-access
/external/libtextclassifier/native/utils/tflite/
Dtoken_encoder.cc110 TfLiteTensor& output_attr = in Prepare() local
112 tflite::SetTensorToDynamic(&output_attr); in Prepare()
Dtext_encoder.cc201 TfLiteTensor& output_attr = in Prepare() local
203 tflite::SetTensorToDynamic(&output_attr); in Prepare()
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator_test.cc693 const auto& output_attr = output_node_def.attr(); in TEST_F() local
694 EXPECT_EQ(2.0, output_attr.at("float_attr").f()); in TEST_F()
695 EXPECT_EQ("Hello World", output_attr.at("str_attr").s()); in TEST_F()
696 EXPECT_EQ(17, output_attr.at("int_attr").i()); in TEST_F()
697 EXPECT_EQ(true, output_attr.at("bool_attr").b()); in TEST_F()
699 const auto& list = output_attr.at("list_string_attr").list(); in TEST_F()
708 const auto& list = output_attr.at("list_float_attr").list(); in TEST_F()
715 const auto& list = output_attr.at("list_int_attr").list(); in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dgraph_compiler.cc152 absl::InlinedVector<AllocatorAttributes, 4> output_attr(n->num_outputs()); in Compile() local
153 params.output_attr_array = output_attr.data(); in Compile()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_with_tf2xla.cc502 llvm::SmallVector<tensorflow::AllocatorAttributes, 4> output_attr( in LegalizeOp() local
504 params_.output_attr_array = output_attr.data(); in LegalizeOp()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.cc468 const auto output_attr = params_->output_attr_array == nullptr in forward_input_to_output_with_shape() local
473 output_shape, output_memory_type(output_index), output_attr); in forward_input_to_output_with_shape()
501 const AllocatorAttributes& output_attr) { in forward_input() argument
556 if (!output_attr.IsEqualOrLessRestrictiveThan(input_attr)) { in forward_input()
Dop_kernel.h894 const AllocatorAttributes& output_attr) TF_MUST_USE_RESULT;
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc819 AttrValue output_attr; in UpdateFunction() local
820 output_attr.mutable_list()->add_shape()->Swap(&proto); in UpdateFunction()
821 (*fun_node->mutable_attr())["_output_shapes"] = output_attr; in UpdateFunction()