HomeSort by relevance Sort by last modified time
    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