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

  /external/tensorflow/tensorflow/core/kernels/
split_op.cc 108 std::tuple<IndexType, IndexType, IndexType> SetDims(
233 Base::template SetDims<Eigen::DenseIndex>(input_shape, split_dim);
299 Base::template SetDims<int32>(input_shape, split_dim);
361 Base::template SetDims<Eigen::DenseIndex>(input_shape, split_dim);
split_v_op.cc 162 std::tuple<IndexType, IndexType, IndexType> SetDims(
288 Base::template SetDims<Eigen::DenseIndex>(input_shape, split_dim);
363 Base::template SetDims<int32>(input_shape, split_dim);
410 Base::template SetDims<Eigen::DenseIndex>(input_shape, split_dim);

Completed in 885 milliseconds