OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:dim_sizes
(Results
1 - 4
of
4
) sorted by null
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
select_op.cc
68
const auto
dim_sizes
= then_shape.
dim_sizes
();
variable
69
absl::Span<const int64> bdims =
dim_sizes
;
split_op.cc
180
auto
dim_sizes
= input_shape.
dim_sizes
();
variable
181
std::vector<int64> limits(
dim_sizes
.begin(),
dim_sizes
.end());
/external/tensorflow/tensorflow/compiler/aot/
codegen.cc
131
string
dim_sizes
, indices;
local
134
dim_sizes
= "[1]";
139
dim_sizes
+= absl::StrCat("[", shape.dimensions(dim), "]");
146
rewrites->push_back({"{{
DIM_SIZES
}}",
dim_sizes
});
188
return (*static_cast<{{TYPE}}(*){{
DIM_SIZES
}}>(
195
return (*static_cast<const {{TYPE}}(*){{
DIM_SIZES
}}>(
231
return (*static_cast<{{TYPE}}(*){{
DIM_SIZES
}}>(
238
return (*static_cast<const {{TYPE}}(*){{
DIM_SIZES
}}>(
277
return (*static_cast<{{TYPE}}(*){{
DIM_SIZES
}}>(
[
all
...]
/external/tensorflow/tensorflow/core/framework/
tensor_shape_test.cc
219
/// REQUIRES: `
dim_sizes
[i] >= 0`
220
explicit TensorShapeOld(gtl::ArraySlice<int64>
dim_sizes
);
221
TensorShapeOld(std::initializer_list<int64>
dim_sizes
)
222
: TensorShapeOld(gtl::ArraySlice<int64>(
dim_sizes
)) {}
276
gtl::ArraySlice<int64>
dim_sizes
() const { return dim_sizes_; }
function in class:tensorflow::__anon45008::TensorShapeOld
390
TensorShapeOld::TensorShapeOld(gtl::ArraySlice<int64>
dim_sizes
) {
391
dim_sizes_.reserve(
dim_sizes
.size());
393
for (auto s :
dim_sizes
) {
Completed in 244 milliseconds