Searched defs:output_attr (Results 1 – 9 of 9) sorted by relevance
56 AllocatorAttributes output_attr; in _mlir_ciface_tf_alloc() local
98 auto output_attr = tf_attrs.get("outputs"); in HandleInputOutputArraysWithModule() local
110 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
502 llvm::SmallVector<tensorflow::AllocatorAttributes, 4> output_attr( in LegalizeOp() local
468 const auto output_attr = params_->output_attr_array == nullptr in forward_input_to_output_with_shape() local501 const AllocatorAttributes& output_attr) { in forward_input()
693 const auto& output_attr = output_node_def.attr(); in TEST_F() local
819 AttrValue output_attr; in UpdateFunction() local