OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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