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