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
69
const auto
dim_sizes
= then_shape.
dim_sizes
();
variable
70
gtl::ArraySlice<int64> bdims =
dim_sizes
;
split_op.cc
195
auto
dim_sizes
= input_shape.
dim_sizes
();
variable
196
std::vector<int64> limits(
dim_sizes
.begin(),
dim_sizes
.end());
/external/tensorflow/tensorflow/compiler/aot/
codegen.cc
118
string
dim_sizes
, indices;
local
121
dim_sizes
= "[1]";
126
dim_sizes
+= strings::StrCat("[", shape.dimensions(dim), "]");
133
rewrites->push_back({"{{
DIM_SIZES
}}",
dim_sizes
});
172
return (*static_cast<{{TYPE}}(*){{
DIM_SIZES
}}>(
179
return (*static_cast<const {{TYPE}}(*){{
DIM_SIZES
}}>(
213
return (*static_cast<{{TYPE}}(*){{
DIM_SIZES
}}>(
220
return (*static_cast<const {{TYPE}}(*){{
DIM_SIZES
}}>(
/external/tensorflow/tensorflow/core/framework/
tensor_shape_test.cc
212
/// REQUIRES: `
dim_sizes
[i] >= 0`
213
explicit TensorShapeOld(gtl::ArraySlice<int64>
dim_sizes
);
214
TensorShapeOld(std::initializer_list<int64>
dim_sizes
)
215
: TensorShapeOld(gtl::ArraySlice<int64>(
dim_sizes
)) {}
269
gtl::ArraySlice<int64>
dim_sizes
() const { return dim_sizes_; }
function in class:tensorflow::__anon39597::TensorShapeOld
383
TensorShapeOld::TensorShapeOld(gtl::ArraySlice<int64>
dim_sizes
) {
384
dim_sizes_.reserve(
dim_sizes
.size());
386
for (auto s :
dim_sizes
) {
Completed in 85 milliseconds