OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:shapes_and_types
(Results
1 - 8
of
8
) sorted by null
/external/tensorflow/tensorflow/core/ops/
data_flow_ops_test.cc
205
std::vector<ShapeInferenceTestOp::ShapeAndType>
shapes_and_types
;
local
206
op.input_resource_handle_shapes_and_types.push_back(&
shapes_and_types
);
210
shapes_and_types
.emplace_back("[1,?,3]", DT_FLOAT);
214
shapes_and_types
.emplace_back("[?,2]", DT_FLOAT);
229
std::vector<ShapeInferenceTestOp::ShapeAndType>
shapes_and_types
;
local
230
op.input_resource_handle_shapes_and_types.push_back(&
shapes_and_types
);
233
shapes_and_types
.emplace_back("[1,?,3]", DT_FLOAT);
236
shapes_and_types
.emplace_back("[?,2]", DT_FLOAT);
245
shapes_and_types
.clear();
248
op.input_resource_handle_shapes_and_types.push_back(&
shapes_and_types
);
279
std::vector<ShapeInferenceTestOp::ShapeAndType>
shapes_and_types
;
local
[
all
...]
array_ops_test.cc
192
const auto*
shapes_and_types
= c.output_handle_shapes_and_types(0);
local
193
ASSERT_TRUE(
shapes_and_types
!= nullptr);
194
ASSERT_EQ(1,
shapes_and_types
->size());
195
EXPECT_EQ((*
shapes_and_types
)[0].dtype, DT_BOOL);
[
all
...]
/external/tensorflow/tensorflow/core/framework/
shape_inference.cc
[
all
...]
shape_inference.h
558
const std::vector<ShapeAndType>&
shapes_and_types
) TF_MUST_USE_RESULT;
563
const std::vector<ShapeAndType>&
shapes_and_types
) TF_MUST_USE_RESULT;
576
const std::vector<ShapeAndType>&
shapes_and_types
) TF_MUST_USE_RESULT;
581
const std::vector<ShapeAndType>&
shapes_and_types
) TF_MUST_USE_RESULT;
596
int idx, const std::vector<ShapeAndType>&
shapes_and_types
) {
598
new std::vector<ShapeAndType>(
shapes_and_types
));
696
const std::vector<ShapeAndType>&
shapes_and_types
,
701
const std::vector<ShapeAndType>&
shapes_and_types
,
shape_inference_test.cc
[
all
...]
/external/tensorflow/tensorflow/core/grappler/costs/
graph_properties.h
75
// Merges shapes <
shapes_and_types
>, determined from an EnqueueV2 node, into
79
const std::vector<shape_inference::ShapeAndType>&
shapes_and_types
,
81
// Relaxes shapes <
shapes_and_types
>, determined from an EnqueueV2 node, into
85
const std::vector<shape_inference::ShapeAndType>&
shapes_and_types
,
graph_properties.cc
641
const std::vector<ShapeAndType>&
shapes_and_types
,
643
if (
shapes_and_types
.size() != queue_shapes_and_types->size()) {
645
"Enqueue nodes mixed number of tensors: ",
shapes_and_types
.size(),
648
for (size_t i = 0; i <
shapes_and_types
.size(); ++i) {
649
const ShapeAndType& a =
shapes_and_types
[i];
664
const std::vector<ShapeAndType>&
shapes_and_types
,
666
if (
shapes_and_types
.size() != queue_shapes_and_types->size()) {
668
"Enqueue nodes mixed number of tensors: ",
shapes_and_types
.size(),
671
for (size_t i = 0; i <
shapes_and_types
.size(); ++i) {
672
const ShapeAndType& a =
shapes_and_types
[i]
893
std::vector<ShapeAndType>
shapes_and_types
;
local
[
all
...]
/external/tensorflow/tensorflow/python/framework/
cpp_shape_inference.cc
142
const auto*
shapes_and_types
= c.output_handle_shapes_and_types(i);
local
143
if (
shapes_and_types
!= nullptr) {
146
for (const auto& p : *
shapes_and_types
) {
Completed in 100 milliseconds