HomeSort by relevance Sort by last modified time
    Searched refs:shape_vector (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/contrib/coder/kernels/
range_coder_ops_test.cc 380 std::vector<int64> shape_vector = {2, 2, 2, 2, 2, 2, 2, 2, 2}; local
381 data = Tensor{DT_INT16, TensorShape{shape_vector}};
389 shape = Tensor{DT_INT32, {static_cast<int64>(shape_vector.size())}};
390 for (int i = 0; i < shape_vector.size(); ++i) {
391 shape.flat<int32>()(i) = shape_vector[i];
  /external/tensorflow/tensorflow/python/framework/
ops.py     [all...]
  /external/tensorflow/tensorflow/core/grappler/optimizers/
layout_optimizer_test.cc 143 TensorShape shape_vector({input_channels});
144 Tensor data_vector(DT_FLOAT, shape_vector);
    [all...]

Completed in 303 milliseconds