OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:reshaped_shape
(Results
1 - 3
of
3
) sorted by null
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
depthtospace_op.cc
72
std::vector<int64>
reshaped_shape
;
variable
75
reshaped_shape
.reserve(input_rank);
79
reshaped_shape
.push_back(input_shape[0]);
81
reshaped_shape
.push_back(input_shape[1 + i]);
85
reshaped_shape
.push_back(block_size_);
88
reshaped_shape
.push_back(input_shape[feature_dim] / block_elems);
104
reshaped_shape
.push_back(input_shape[0]);
107
reshaped_shape
.push_back(block_size_);
110
reshaped_shape
.push_back(input_shape[feature_dim] / block_elems);
112
reshaped_shape
.push_back(input_shape[2 + i])
[
all
...]
spacetodepth_op.cc
72
std::vector<int64>
reshaped_shape
;
variable
75
reshaped_shape
.reserve(input_rank);
88
reshaped_shape
.push_back(input_shape[0]);
90
reshaped_shape
.push_back(input_shape[1 + i] / block_size_);
91
reshaped_shape
.push_back(block_size_);
93
reshaped_shape
.push_back(input_shape[feature_dim]);
120
reshaped_shape
.push_back(input_shape[0]);
121
reshaped_shape
.push_back(input_shape[feature_dim]);
123
reshaped_shape
.push_back(input_shape[2 + i] / block_size_);
124
reshaped_shape
.push_back(block_size_)
[
all
...]
/external/tensorflow/tensorflow/compiler/xla/service/
algebraic_simplifier_test.cc
4896
Shape
reshaped_shape
= ShapeUtil::MakeShape(F32, {0});
local
[
all
...]
Completed in 1061 milliseconds