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 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