Searched refs:compute_transposed_shape (Results 1 – 7 of 7) sorted by relevance
56 const TensorShape dst_shape = misc::shape_calculator::compute_transposed_shape(*src); in configure()93 … dst_info = src->clone()->set_tensor_shape(misc::shape_calculator::compute_transposed_shape(*src)); in validate()
223 TensorShape cell_state1_shape = compute_transposed_shape(*recurrent_to_output_weights->info()); in configure()455 …TensorShape units_out_transposed_shape = compute_transposed_shape(*recurrent_to_output_weight… in validate()456 TensorShape num_units_transposed_shape = compute_transposed_shape(*forget_gate_bias); in validate()
244 TensorShape cell_state1_shape = compute_transposed_shape(*recurrent_to_output_weights->info()); in configure()480 …TensorShape units_out_transposed_shape = compute_transposed_shape(*recurrent_to_output_weight… in validate()481 TensorShape num_units_transposed_shape = compute_transposed_shape(*forget_gate_bias); in validate()
432 const TensorShape dst_shape = misc::shape_calculator::compute_transposed_shape(*src); in configure()468 const TensorShape dst_shape = misc::shape_calculator::compute_transposed_shape(*src); in validate()
343 …lone()->set_is_resizable(true).reset_padding().set_tensor_shape(compute_transposed_shape(*weights)… in validate()
384 …lone()->set_is_resizable(true).reset_padding().set_tensor_shape(compute_transposed_shape(*weights)… in validate()
403 inline TensorShape compute_transposed_shape(const ITensorInfo &input) in compute_transposed_shape() function