OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:tensor_shape_map
(Results
1 - 4
of
4
) sorted by null
/external/tensorflow/tensorflow/core/kernels/hexagon/
graph_transfer_utils.cc
61
const RemoteFusedGraphExecuteUtils::TensorShapeMap&
tensor_shape_map
) {
81
RemoteFusedGraphExecuteUtils::GetTensorShapeType(
tensor_shape_map
,
103
RemoteFusedGraphExecuteUtils::TensorShapeMap
tensor_shape_map
;
local
105
*original_def, inputs, true /* initialize_by_zero */, &
tensor_shape_map
);
108
AddOutputTensorShapeTypeByTensorShapeMap(
tensor_shape_map
, &node_def));
131
tensor_shape_map
);
137
RemoteFusedGraphExecuteUtils::GetTensorShapeType(
tensor_shape_map
,
graph_transferer.cc
215
RemoteFusedGraphExecuteUtils::TensorShapeMap
tensor_shape_map
;
local
217
graph_def, input_node_info_list, true, &
tensor_shape_map
);
222
TF_CHECK_OK(AddOutputTensorShapeTypeByTensorShapeMap(
tensor_shape_map
,
[
all
...]
/external/tensorflow/tensorflow/core/kernels/
remote_fused_graph_execute_utils_test.cc
89
RemoteFusedGraphExecuteUtils::TensorShapeMap
tensor_shape_map
;
local
298
RemoteFusedGraphExecuteUtils::TensorShapeMap
tensor_shape_map
;
local
306
def, inputs, false /* initialize_by_zero */, &
tensor_shape_map
);
311
ASSERT_EQ(3,
tensor_shape_map
.size());
312
ASSERT_EQ(1,
tensor_shape_map
.count(NAME_A));
313
ASSERT_EQ(1,
tensor_shape_map
.count(NAME_B));
314
ASSERT_EQ(1,
tensor_shape_map
.count(NAME_A_PLUS_B));
317
RemoteFusedGraphExecuteUtils::GetTensorShapeType(
tensor_shape_map
,
323
tst = RemoteFusedGraphExecuteUtils::GetTensorShapeType(
tensor_shape_map
,
342
RemoteFusedGraphExecuteUtils::TensorShapeMap
tensor_shape_map
;
local
[
all
...]
remote_fused_graph_execute_utils.cc
269
RemoteFusedGraphExecuteUtils::TensorShapeMap*
tensor_shape_map
) {
270
CHECK(
tensor_shape_map
!= nullptr);
310
EmplaceTensorShapeType(name, tensor,
tensor_shape_map
);
315
EmplaceTensorShapeType(name, tensor,
tensor_shape_map
);
338
TensorShapeMap*
tensor_shape_map
) {
339
CHECK_NE(
tensor_shape_map
, nullptr);
340
tensor_shape_map
->clear();
341
tensor_shape_map
->reserve(input_node_info_list.size() +
348
EmplaceTensorShapeType(node_name, tensor,
tensor_shape_map
);
374
const TensorShapeMap&
tensor_shape_map
, NodeDef* node_def)
579
TensorShapeMap
tensor_shape_map
;
local
[
all
...]
Completed in 659 milliseconds