HomeSort by relevance Sort by last modified time
    Searched refs:output_attr (Results 1 - 7 of 7) sorted by null

  /external/tensorflow/tensorflow/python/ops/
functional_ops.py 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-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
    [all...]
  /external/libtextclassifier/utils/tflite/
token_encoder.cc 110 TfLiteTensor& output_attr = local
112 tflite::SetTensorToDynamic(&output_attr);
text_encoder.cc 200 TfLiteTensor& output_attr = local
202 tflite::SetTensorToDynamic(&output_attr);
  /external/tensorflow/tensorflow/compiler/tf2xla/
graph_compiler.cc 146 absl::InlinedVector<AllocatorAttributes, 4> output_attr(n->num_outputs());
147 params.output_attr_array = output_attr.data();
  /external/tensorflow/tensorflow/lite/toco/tflite/
operator_test.cc 627 const auto& output_attr = output_node_def.attr(); local
628 EXPECT_EQ(2.0, output_attr.at("float_attr").f());
629 EXPECT_EQ("Hello World", output_attr.at("str_attr").s());
630 EXPECT_EQ(17, output_attr.at("int_attr").i());
631 EXPECT_EQ(true, output_attr.at("bool_attr").b());
633 const auto& list = output_attr.at("list_string_attr").list();
642 const auto& list = output_attr.at("list_float_attr").list();
649 const auto& list = output_attr.at("list_int_attr").list();
    [all...]
  /external/tensorflow/tensorflow/core/framework/
op_kernel.cc 452 const auto output_attr = params_->output_attr_array == nullptr local
457 output_shape, output_memory_type(output_index), output_attr);
499 const AllocatorAttributes& output_attr) {
554 if (!output_attr.IsEqualOrLessRestrictiveThan(input_attr)) {
    [all...]
op_kernel.h     [all...]

Completed in 195 milliseconds