Searched defs:output_attr (Results 1 – 9 of 9) sorted by relevance
104 auto output_attr = tf_attrs.get("outputs"); in HandleInputOutputArraysWithModule() local
110 TfLiteTensor& output_attr = in Prepare() local
135 TfLiteTensor& output_attr = in Prepare() local
201 TfLiteTensor& output_attr = in Prepare() local
152 absl::InlinedVector<AllocatorAttributes, 4> output_attr(n->num_outputs()); in Compile() local
642 llvm::SmallVector<tensorflow::AllocatorAttributes, 4> output_attr( in LegalizeOp() local
481 const auto output_attr = params_->output_attr_array == nullptr in forward_input_to_output_with_shape() local514 const AllocatorAttributes& output_attr) { in forward_input()
695 const auto& output_attr = output_node_def.attr(); in TEST_F() local
848 AttrValue output_attr; in UpdateFunction() local