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

  /external/tensorflow/tensorflow/compiler/tf2xla/
graph_compiler.cc 122 gtl::InlinedVector<AllocatorAttributes, 4> output_attr(n->num_outputs());
123 params.output_attr_array = output_attr.data();
  /external/tensorflow/tensorflow/contrib/lite/toco/tflite/
operator_test.cc 409 const auto& output_attr = output_node_def.attr(); local
410 EXPECT_EQ(2.0, output_attr.at("float_attr").f());
411 EXPECT_EQ("Hello World", output_attr.at("str_attr").s());
412 EXPECT_EQ(17, output_attr.at("int_attr").i());
413 EXPECT_EQ(true, output_attr.at("bool_attr").b());
  /external/tensorflow/tensorflow/core/framework/
op_kernel.cc 414 const auto output_attr = params_->output_attr_array == nullptr local
419 output_memory_type(output_index), output_attr);
460 MemoryType output_memory_type, const AllocatorAttributes& output_attr) {
486 if (!output_attr.IsEqualOrLessRestrictiveThan(input_attr)) {
    [all...]

Completed in 779 milliseconds