Searched refs:reshaped_inputs (Results 1 – 2 of 2) sorted by relevance
72 std::vector<xla::XlaOp> reshaped_inputs(num); in Compile() local79 reshaped_inputs[i] = xla::Reshape(values[i], child_shape.dim_sizes()); in Compile()82 ctx->SetOutput(0, xla::ConcatInDim(ctx->builder(), reshaped_inputs, axis)); in Compile()
124 reshaped_inputs = []135 reshaped_inputs.append(x)136 return self._merge_function(reshaped_inputs)155 reshaped_inputs.append(x_transposed)159 reshaped_inputs.append(array_ops.transpose(x, perm=dims))163 reshaped_inputs.append(x)164 y = self._merge_function(reshaped_inputs)