Searched refs:tensor_in_and_out_dims (Results 1 – 2 of 2) sorted by relevance
76 constexpr int tensor_in_and_out_dims = 4; in Compute() local79 OP_REQUIRES(context, tensor_in.dims() == tensor_in_and_out_dims, in Compute()82 std::vector<int> input_size(tensor_in_and_out_dims); in Compute()83 std::vector<int> output_size(tensor_in_and_out_dims); in Compute()84 for (int i = 0; i < tensor_in_and_out_dims; ++i) { in Compute()88 for (int i = 0; i < tensor_in_and_out_dims; ++i) { in Compute()237 constexpr int tensor_in_and_out_dims = 4; in Compute() local238 std::vector<int64> input_size(tensor_in_and_out_dims); in Compute()239 std::vector<int64> output_size(tensor_in_and_out_dims); in Compute()240 for (int i = 0; i < tensor_in_and_out_dims; ++i) { in Compute()[all …]
73 constexpr int tensor_in_and_out_dims = 4; in Compute() local76 OP_REQUIRES(context, tensor_in.dims() == tensor_in_and_out_dims, in Compute()79 std::vector<int> input_size(tensor_in_and_out_dims); in Compute()80 std::vector<int> output_size(tensor_in_and_out_dims); in Compute()81 for (int i = 0; i < tensor_in_and_out_dims; ++i) { in Compute()85 for (int i = 0; i < tensor_in_and_out_dims; ++i) { in Compute()258 constexpr int tensor_in_and_out_dims = 4; in Compute() local261 for (auto i = 0; i < tensor_in_and_out_dims; ++i) { in Compute()